| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | specify_graph_input_format.cc | 31 if (format_ != mindspore::NCHW) { in Run()
67 if (format_ == mindspore::NCHW) { in HandleGraphInput()
73 if (format_ == mindspore::NCHW) { in HandleGraphInput()
84 if (format_ == mindspore::NCHW) { in HandleGraphInput()
85 trans_prim->AddAttr(ops::kFormat, MakeValue<int64_t>(NCHW)); in HandleGraphInput()
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | batchnorm.py | 22 @VLD.add_format(DF.NCHW, DF.DEFAULT, DF.DEFAULT, DF.DEFAULT, DF.DEFAULT)
52 if input_x.data_format in (DF.DEFAULT, DF.NCHW):
63 if input_x.data_format in (DF.DEFAULT, DF.NCHW):
97 if input_x.data_format in (DF.DEFAULT, DF.NCHW):
116 if input_x.data_format in (DF.DEFAULT, DF.NCHW):
122 if input_x.data_format in (DF.DEFAULT, DF.NCHW):
128 if input_x.data_format in (DF.DEFAULT, DF.NCHW):
|
| D | batchnorm_grad.py | 22 @VLD.add_format(DF.NCHW, DF.NCHW, DF.DEFAULT, DF.DEFAULT, DF.DEFAULT, DF.DEFAULT)
65 if input_x.data_format in (DF.DEFAULT, DF.NCHW):
81 if input_x.data_format in (DF.DEFAULT, DF.NCHW):
|
| D | bias_add_grad.py | 22 @VLD.add_format(DF.NCHW)
33 elif x.data_format == DF.NCHW:
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_pool_parser.cc | 50 if (format == mindspore::Format::NCHW) { in Parse()
60 if (format == mindspore::Format::NCHW) { in Parse()
95 if (format == mindspore::Format::NCHW) { in Parse()
105 if (format == mindspore::Format::NCHW) { in Parse()
|
| D | tf_util.cc | 147 return mindspore::Format::NCHW; in ParseNodeFormat()
152 return mindspore::Format::NCHW; in ParseNodeFormat()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | to_format_base.cc | 52 trans_prim->AddAttr(ops::kFormat, MakeValue<int64_t>(NCHW)); in GenNewInput()
98 if (format_ == mindspore::NCHW) { in ModifyCNode()
225 if (format_ == mindspore::NCHW) { in HandleGraphInput()
231 if (format_ == mindspore::NCHW) { in HandleGraphInput()
242 if (format_ == mindspore::NCHW) { in HandleGraphInput()
243 trans_prim->AddAttr(ops::kFormat, MakeValue<int64_t>(NCHW)); in HandleGraphInput()
392 if (format_ != mindspore::NHWC && format_ != mindspore::NCHW) { in Run()
|
| D | to_nhwc_format.cc | 32 if (node_format != mindspore::NCHW) { in GetTransNodeFormatType()
55 if (node_format != mindspore::NCHW) { in DecideConvWeightSrcAndDstFormat()
|
| D | to_nchw_format.cc | 28 if (node_format == mindspore::NCHW) { in GetTransNodeFormatType()
52 if (node_format == mindspore::NCHW) { in DecideConvWeightSrcAndDstFormat()
|
| D | to_nchw_format.h | 28 format_ = mindspore::NCHW;
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_max_pool_grad.cc | 36 max_pool_grad->Init({1}, {1}, VALID, NCHW); in TEST_F()
38 EXPECT_EQ(max_pool_grad->get_data_format(), NCHW); in TEST_F()
|
| D | test_ops_pooling_grad.cc | 37 std::vector<int64_t>{1, 1, 1, 1}, FLOOR, NCHW, false); in TEST_F()
44 EXPECT_EQ(pooling_grad->get_format(), NCHW); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | fullconnection_fp32_tests.cc | 108 …auto *in_t = new Tensor(kNumberTypeFloat, {20, 4, 2, 10}, mindspore::NCHW, lite::Tensor::Category:… in FcTestInit2()
115 …auto *weight_t = new Tensor(kNumberTypeFloat, {30, 80}, mindspore::NCHW, lite::Tensor::Category::C… in FcTestInit2()
122 …auto *bias_t = new Tensor(kNumberTypeFloat, {30}, mindspore::NCHW, lite::Tensor::Category::CONST_T… in FcTestInit2()
129 …auto *out_t = new Tensor(kNumberTypeFloat, {20, 30}, mindspore::NCHW, lite::Tensor::Category::CONS… in FcTestInit2()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | pool_tensorrt.cc | 37 if (in_tensors[0].format() != Format::NHWC && in_tensors[0].format() != Format::NCHW) { in IsSupport()
93 this->AddInnerOutTensors(ITensorHelper{activation_layer->getOutput(0), Format::NCHW}); in AddInnerOp()
94 MS_LOG(DEBUG) << "output " << GetTensorFormat(activation_layer->getOutput(0), Format::NCHW); in AddInnerOp()
|
| D | concate_tensorrt.cc | 70 int transpose_tensor_index = tensorrt_in_tensors_[0].format_ == Format::NCHW ? 0 : 1; in AddInnerOp()
88 if (trt_input_tensors[0]->getDimensions().nbDims == DIMENSION_4D && out_format == Format::NCHW) { in AddInnerOp()
|
| D | pad_tensorrt.cc | 52 if (in_tensors[0].format() != Format::NHWC && in_tensors[0].format() != Format::NCHW) { in IsSupport()
109 this->AddInnerOutTensors(ITensorHelper{padding_layer->getOutput(0), Format::NCHW}); in AddInnerOp()
|
| D | convolution_tensorrt.cc | 38 if (in_tensors[0].format() != Format::NHWC && in_tensors[0].format() != Format::NCHW) { in IsSupport()
123 this->AddInnerOutTensors(ITensorHelper{activation_layer->getOutput(0), Format::NCHW}); in AddInnerOp()
|
| D | shuffle_tensorrt.cc | 80 if (tensorrt_in_tensors_[0].format_ == Format::NCHW) { in AddInnerOp()
95 out_format_ = Format::NCHW; in AddInnerOp()
236 out_format_ = Format::NCHW; in AddTransposeOp()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | tensor.cc | 102 case mindspore::NCHW: in Batch()
128 case mindspore::NCHW: in Channel()
155 case mindspore::NCHW: in Height()
182 case mindspore::NCHW: in Width()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/include/ir/ |
| D | format.h | 26 NCHW = 0, enumerator
|
| /third_party/mindspore/include/api/ |
| D | format.h | 26 NCHW = 0, enumerator
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | batch_norm_grad_cpu_kernel.cc | 48 (void)x_shape.insert(x_shape.end(), (NCHW - NC), 1); in InitKernel()
49 } else if (x_shape.size() != NCHW) { in InitKernel()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | bias_add.h | 41 void Init(const Format &format = NCHW);
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_pooling_parser.cc | 137 prim->AddAttr(mindspore::ops::kOriginalFormat, MakeValue<int64_t>(mindspore::Format::NCHW)); in Parse()
150 prim->AddAttr(mindspore::ops::kOriginalFormat, MakeValue<int64_t>(mindspore::Format::NCHW)); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_pool_parser.cc | 30 prim->AddAttr(mindspore::ops::kOriginalFormat, MakeValue<int64_t>(mindspore::Format::NCHW)); in Parse()
104 prim->AddAttr(mindspore::ops::kOriginalFormat, MakeValue<int64_t>(mindspore::Format::NCHW)); in Parse()
|