ONE - On-device Neural Engine
|
#include <Attributes.h>
Public Member Functions | |
Conv2DOpAttributes ()=default | |
Data Fields | |
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} |
std::int32_t | num_groups {1} |
DataFormat | data_format {DataFormat::NHWC} |
Definition at line 27 of file Attributes.h.
|
default |
DataFormat mir::Conv2DOpAttributes::data_format {DataFormat::NHWC} |
Definition at line 35 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertConv(), mir_caffe::CaffeOpCreator::convertConvolution(), mir_onnx::convertConvV1(), mir::ops::Conv2DOp::getDataFormat(), mir::ops::DepthwiseConv2DOp::getDataFormat(), mir_interpreter::Conv2DImpl< T >::run(), and mir_interpreter::Conv2DImpl< uint8_t >::run().
std::int32_t mir::Conv2DOpAttributes::num_groups {1} |
Definition at line 34 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertConv(), mir_caffe::CaffeOpCreator::convertConvolution(), mir_onnx::convertConvV1(), mir::ops::Conv2DOp::getNumGroups(), mir_interpreter::Conv2DImpl< T >::run(), and mir_interpreter::Conv2DImpl< uint8_t >::run().
std::vector<std::int32_t> mir::Conv2DOpAttributes::padding_after {0, 0} |
Definition at line 33 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertConv(), mir_tflite::TFLiteOpCreator::convertConv2D(), mir_onnx::convertConvV1(), mir_tflite::TFLiteOpCreator::convertDepthwiseConv2D(), mir::ops::Conv2DOp::getPaddingAfter(), and mir::ops::DepthwiseConv2DOp::getPaddingAfter().
std::vector<std::int32_t> mir::Conv2DOpAttributes::padding_before {0, 0} |
Definition at line 32 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertConv(), mir_tflite::TFLiteOpCreator::convertConv2D(), mir_onnx::convertConvV1(), mir_tflite::TFLiteOpCreator::convertDepthwiseConv2D(), mir::ops::Conv2DOp::getPaddingBefore(), mir::ops::DepthwiseConv2DOp::getPaddingBefore(), mir_interpreter::Conv2DImpl< T >::run(), and mir_interpreter::Conv2DImpl< uint8_t >::run().
std::vector<std::int32_t> mir::Conv2DOpAttributes::strides {1, 1} |
Definition at line 31 of file Attributes.h.
Referenced by mir_caffe2::Caffe2OpCreator::convertConv(), mir_tflite::TFLiteOpCreator::convertConv2D(), mir_onnx::convertConvV1(), mir_tflite::TFLiteOpCreator::convertDepthwiseConv2D(), mir::ops::Conv2DOp::getStrides(), mir::ops::DepthwiseConv2DOp::getStrides(), mir_interpreter::Conv2DImpl< T >::run(), mir_interpreter::Conv2DImpl< uint8_t >::run(), TEST(), TEST(), TEST(), and TEST().