ONE - On-device Neural Engine
|
#include <Attributes.h>
Public Member Functions | |
AvgPool2DOpAttributes ()=default | |
Data Fields | |
std::vector< std::int32_t > | window {1, 1} |
std::vector< std::int32_t > | strides {1, 1} |
std::vector< std::int32_t > | padding_before {0, 0} |
std::vector< std::int32_t > | padding_after {0, 0} |
DataFormat | data_format {DataFormat::NHWC} |
bool | include_pad {true} |
Definition at line 38 of file Attributes.h.
|
default |
DataFormat mir::AvgPool2DOpAttributes::data_format {DataFormat::NHWC} |
Definition at line 46 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertAveragePool(), mir_onnx::convertAveragePoolV1(), mir_onnx::convertGlobalAveragePoolV2(), mir_caffe::CaffeOpCreator::convertPooling(), and mir::ops::AvgPool2DOp::getDataFormat().
bool mir::AvgPool2DOpAttributes::include_pad {true} |
Definition at line 47 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertAveragePool(), mir_tflite::TFLiteOpCreator::convertAveragePool2D(), mir_onnx::convertAveragePoolV1(), mir::ops::AvgPool2DOp::getIncludePad(), and TEST().
std::vector<std::int32_t> mir::AvgPool2DOpAttributes::padding_after {0, 0} |
Definition at line 45 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertAveragePool(), mir_tflite::TFLiteOpCreator::convertAveragePool2D(), mir_onnx::convertAveragePoolV1(), and mir::ops::AvgPool2DOp::getPaddingAfter().
std::vector<std::int32_t> mir::AvgPool2DOpAttributes::padding_before {0, 0} |
Definition at line 44 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertAveragePool(), mir_tflite::TFLiteOpCreator::convertAveragePool2D(), mir_onnx::convertAveragePoolV1(), and mir::ops::AvgPool2DOp::getPaddingBefore().
std::vector<std::int32_t> mir::AvgPool2DOpAttributes::strides {1, 1} |
Definition at line 43 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertAveragePool(), mir_tflite::TFLiteOpCreator::convertAveragePool2D(), mir_onnx::convertAveragePoolV1(), mir::ops::AvgPool2DOp::getStrides(), and TEST().
std::vector<std::int32_t> mir::AvgPool2DOpAttributes::window {1, 1} |
Definition at line 42 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertAveragePool(), mir_tflite::TFLiteOpCreator::convertAveragePool2D(), mir_onnx::convertAveragePoolV1(), mir_onnx::convertGlobalAveragePoolV2(), mir::ops::AvgPool2DOp::getWindowSize(), TEST(), and TEST().