| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | pad_fusion.cc | 26 void PadFusion::Init(const PaddingMode &padding_mode, const float constant_value) { in Init()
31 void PadFusion::set_padding_mode(const PaddingMode &padding_mode) { in set_padding_mode()
40 PaddingMode PadFusion::get_padding_mode() const { in get_padding_mode()
43 return PaddingMode(GetValue<int64_t>(value_ptr)); in get_padding_mode()
|
| D | pad_fusion.h | 41 void Init(const PaddingMode &padding_mode, const float constant_value);
46 void set_padding_mode(const PaddingMode &padding_mode);
56 PaddingMode get_padding_mode() const;
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | pad_fp32_test.cc | 24 using mindspore::schema::PaddingMode;
34 … float *output_data, PaddingMode mode, int *paddings, int padding_length, float constant_value,
61 … float *output_data, PaddingMode mode, int *paddings, int padding_length, float constant_value, in Prepare()
105 PaddingMode mode = PaddingMode_SYMMETRIC; in TEST_F()
147 PaddingMode mode = PaddingMode_REFLECT; in TEST_F()
182 PaddingMode mode = PaddingMode_CONSTANT; in TEST_F()
218 PaddingMode mode = PaddingMode_CONSTANT; in TEST_F()
254 PaddingMode mode = PaddingMode_CONSTANT; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_pad_parser.cc | 28 mindspore::PaddingMode padding_mode; in Parse()
44 padding_mode = mindspore::PaddingMode::CONSTANT; in Parse()
46 padding_mode = mindspore::PaddingMode::REFLECT; in Parse()
48 padding_mode = mindspore::PaddingMode::SYMMETRIC; in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_pad_parser.cc | 36 prim->set_padding_mode(mindspore::PaddingMode::CONSTANT); in Parse()
39 prim->set_padding_mode(mindspore::PaddingMode::CONSTANT); in Parse()
68 prim->set_padding_mode(mindspore::PaddingMode::SYMMETRIC); in Parse()
70 prim->set_padding_mode(mindspore::PaddingMode::REFLECT); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_pad_parser.cc | 47 prim->set_padding_mode(mindspore::PaddingMode::CONSTANT); in Parse()
59 prim->set_padding_mode(mindspore::PaddingMode::CONSTANT); in Parse()
81 prim->set_padding_mode(mindspore::PaddingMode::REFLECT); in Parse()
84 prim->set_padding_mode(mindspore::PaddingMode::SYMMETRIC); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | pad_fusion_mapper.cc | 66 {PaddingMode::CONSTANT, "constant"}, in AdjustPadAttr()
67 {PaddingMode::REFLECT, "reflect"}, in AdjustPadAttr()
68 {PaddingMode::SYMMETRIC, "edge"}, in AdjustPadAttr()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | pad_tensorrt.cc | 47 schema::PaddingMode padding_mode = pad_primitive->padding_mode(); in IsSupport()
48 if (padding_mode != schema::PaddingMode::PaddingMode_CONSTANT) { in IsSupport()
|
| D | convolution_tensorrt.cc | 148 conv_layer->setPaddingMode(nvinfer1::PaddingMode::kSAME_UPPER); in SetAttributes()
|
| D | deconvolution_tensorrt.cc | 149 decon_layer->setPaddingMode(nvinfer1::PaddingMode::kSAME_UPPER); in SetAttributes()
|
| D | pool_tensorrt.cc | 178 pooling_layer->setPaddingMode(nvinfer1::PaddingMode::kSAME_UPPER); in AddParams()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops_types_generated.h | 471 enum PaddingMode : int8_t { enum
480 inline const PaddingMode (&EnumValuesPaddingMode())[4] { in EnumValuesPaddingMode()
481 static const PaddingMode values[] = { in EnumValuesPaddingMode()
501 inline const char *EnumNamePaddingMode(PaddingMode e) { in EnumNamePaddingMode()
|
| D | ops_types.fbs | 119 enum PaddingMode : byte {
|
| D | ops_v0.fbs | 122 enum PaddingMode : byte {
137 paddingMode: PaddingMode;
|
| D | ops.fbs | 766 padding_mode: PaddingMode;
|
| D | ops_v0_generated.h | 1084 enum PaddingMode : int8_t { enum
1093 inline const PaddingMode (&EnumValuesPaddingMode())[4] { in EnumValuesPaddingMode()
1094 static const PaddingMode values[] = { in EnumValuesPaddingMode()
1114 inline const char *EnumNamePaddingMode(PaddingMode e) { in EnumNamePaddingMode()
1211 mindspore::schema::v0::PaddingMode paddingMode() const { in paddingMode()
1212 return static_cast<mindspore::schema::v0::PaddingMode>(GetField<int8_t>(VT_PADDINGMODE, 0)); in paddingMode()
1234 void add_paddingMode(mindspore::schema::v0::PaddingMode paddingMode) { in add_paddingMode()
1254 mindspore::schema::v0::PaddingMode paddingMode = mindspore::schema::v0::PaddingMode_CONSTANT,
1266 mindspore::schema::v0::PaddingMode paddingMode = mindspore::schema::v0::PaddingMode_CONSTANT,
|
| /third_party/mindspore/mindspore/lite/mindir/include/ |
| D | mindir_types.h | 185 enum PaddingMode : int8_t { enum
|
| D | mindir.h | 259 PaddingMode padding_mode, float constant_value);
262 PaddingMode MindIR_PadFusion_GetPaddingMode(ConstPrimitivePtr primitive);
263 void MindIR_PadFusion_SetPaddingMode(PrimitivePtr *primitive, PaddingMode padding_mode);
|
| /third_party/mindspore/mindspore/lite/schema/inner/ |
| D | ops_types_generated.h | 480 enum PaddingMode : int8_t { enum
489 inline const PaddingMode (&EnumValuesPaddingMode())[4] { in EnumValuesPaddingMode()
490 static const PaddingMode values[] = { in EnumValuesPaddingMode()
510 inline const char *EnumNamePaddingMode(PaddingMode e) { in EnumNamePaddingMode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 154 layer->setPaddingMode(nvinfer1::PaddingMode::kSAME_UPPER); in AddPoolingLayer()
284 layer->setPaddingMode(nvinfer1::PaddingMode::kSAME_UPPER); in MS_TRT_CONVERTER_FUNC_REG()
737 layer->setPaddingMode(nvinfer1::PaddingMode::kSAME_UPPER); in MS_TRT_CONVERTER_FUNC_REG()
742 layer->setPaddingMode(nvinfer1::PaddingMode::kEXPLICIT_ROUND_DOWN); in MS_TRT_CONVERTER_FUNC_REG()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_pad_fusion.cc | 139 if (pad_mode != PaddingMode::CONSTANT) { in IsPrimitiveProper()
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir.cc | 2580 PaddingMode padding_mode, float constant_value) { in MindIR_PadFusion_CreatePrimitive()
2583 … static_cast<schema::PaddingMode>(padding_mode), constant_value); in MindIR_PadFusion_CreatePrimitive()
2622 … static_cast<schema::PaddingMode>(value->padding_mode()), value->constant_value()); in MindIR_PadFusion_SetPaddings()
2632 PaddingMode MindIR_PadFusion_GetPaddingMode(ConstPrimitivePtr primitive) { in MindIR_PadFusion_GetPaddingMode()
2637 return static_cast<PaddingMode>(value->padding_mode()); in MindIR_PadFusion_GetPaddingMode()
2639 PaddingMode en = static_cast<PaddingMode>(0); in MindIR_PadFusion_GetPaddingMode()
2643 PaddingMode en = static_cast<PaddingMode>(0); in MindIR_PadFusion_GetPaddingMode()
2648 void MindIR_PadFusion_SetPaddingMode(PrimitivePtr *primitive, PaddingMode padding_mode) { in MindIR_PadFusion_SetPaddingMode()
2656 … static_cast<schema::PaddingMode>(padding_mode), value->constant_value()); in MindIR_PadFusion_SetPaddingMode()
2688 … static_cast<schema::PaddingMode>(value->padding_mode()), constant_value); in MindIR_PadFusion_SetConstantValue()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.h | 118 enum PaddingMode : int64_t { CONSTANT = 0, REFLECT = 1, SYMMETRIC = 2, MODE_RESERVED = 3 }; enum
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/ |
| D | opencl_fusion.cc | 235 if (pad_param->pad_mode_ != schema::PaddingMode::PaddingMode_CONSTANT || in TryMergePadXxx()
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_def.cc | 766 OP_ATTR_ENUM(padding_mode, PaddingMode)
|