| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/ |
| D | merge_cast_to_op_test.cc | 51 builder.SetInputsFormat({kOpFormat_NCHW}); in Query()
59 builder.SetOutputsFormat({kOpFormat_NCHW}); in Query()
97 builder1.SetInputsFormat({kOpFormat_NCHW}); in TEST_F()
110 builder2.SetInputsFormat({kOpFormat_NCHW}); in TEST_F()
111 builder2.SetOutputsFormat({kOpFormat_NCHW}); in TEST_F()
160 builder1.SetInputsFormat({kOpFormat_NCHW}); in TEST_F()
161 builder1.SetOutputsFormat({kOpFormat_NCHW}); in TEST_F()
174 builder2.SetOutputsFormat({kOpFormat_NCHW}); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/common/ |
| D | trans_test.cc | 39 FormatArgs format_args{data, device_size, kOpFormat_NCHW, kOpFormat_HWCN, in TEST_F()
56 FormatArgs format_args{data, device_size, kOpFormat_NCHW, kOpFormat_HWCN, in TEST_F()
71 FormatArgs format_args{data, device_size, kOpFormat_NCHW, kOpFormat_NHWC, in TEST_F()
87 FormatArgs format_args{data, device_size, kOpFormat_NCHW, kOpFormat_NHWC, in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/ |
| D | mem_reuse_test.cc | 60 tensor::DeviceInfo device_info{kOpFormat_NCHW, tensor_type}; in CreateKernelGraph()
94 builder.SetInputsFormat({kOpFormat_NCHW, kOpFormat_NCHW}); in CreateKernelGraph()
96 builder.SetOutputsFormat({kOpFormat_NCHW}); in CreateKernelGraph()
124 relu_builder.SetInputsFormat({kOpFormat_NCHW}); in CreateKernelGraph()
125 relu_builder.SetOutputsFormat({kOpFormat_NCHW}); in CreateKernelGraph()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | transdata_split.cc | 25 {kOpFormat_C1HWNCoC0, kOpFormat_NCHW}, {kOpFormat_NCHW, kOpFormat_C1HWNCoC0},
74 if (output_format == kOpFormat_DEFAULT || output_format == kOpFormat_NCHW) { in DoSplit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/expanders/ |
| D | bias_add.cc | 32 support_format->AddFormat({kOpFormat_NCHW, kOpFormat_DEFAULT}); in BiasAdd()
45 if (input_x->format == kOpFormat_NCHW) { in Expand()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | insert_format_transform_op.cc | 34 if ((src_format == kOpFormat_NCHW) && (dst_format == kOpFormat_NHWC)) { in TransposeAxis()
36 } else if ((src_format == kOpFormat_NHWC) && (dst_format == kOpFormat_NCHW)) { in TransposeAxis()
134 origin_data_format = kOpFormat_NCHW; in Process()
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 257 builder.SetOutputsFormat({kOpFormat_NCHW, kOpFormat_NC1HWC0}); in TEST_F()
259 EXPECT_EQ(AnfAlgo::GetOutputFormat(add, 0), kOpFormat_NCHW); in TEST_F()
276 builder.SetInputsFormat({kOpFormat_NCHW, kOpFormat_NC1HWC0}); in TEST_F()
278 EXPECT_EQ(AnfAlgo::GetInputFormat(add, 0), kOpFormat_NCHW); in TEST_F()
295 builder.SetOutputsFormat({kOpFormat_NCHW}); in TEST_F()
299 EXPECT_EQ(AnfAlgo::GetPrevNodeOutputFormat(add, 0), kOpFormat_NCHW); in TEST_F()
374 builder.SetOutputsFormat({kOpFormat_NCHW, kOpFormat_NCHW, kOpFormat_NHWC, kOpFormat_FRAC_NZ}); in TEST_F()
407 builder.SetInputsFormat({kOpFormat_NCHW, kOpFormat_NCHW, kOpFormat_NHWC}); in TEST_F()
462 builder.SetOutputsFormat({kOpFormat_NCHW}); in TEST_F()
479 builder.SetInputsFormat({kOpFormat_NCHW, kOpFormat_NC1HWC0}); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.cc | 208 if (format_ == kOpFormat_NCHW || format_ == kOpFormat_DEFAULT || format_ == kOpFormat_NCDHW) { in SyncDeviceToHost()
275 const std::string dst_format = kOpFormat_NCHW; in SyncDeviceToHostAndConvertFormatBasedOnTransData()
333 const trans::FormatArgs format_args{host_tmp.data(), size_, kOpFormat_NCHW, format_, in SyncDeviceToHostAndConvertFormat()
349 const trans::FormatArgs format_args{host_tmp.data(), size_, kOpFormat_NCHW, format_, in SyncDeviceToHostAndConvertFormat()
375 …if (format_ == kOpFormat_NCHW || format_ == kOpFormat_DEFAULT || format_ == kOpFormat_NCDHW || for… in SyncHostToDevice()
457 const trans::FormatArgs format_args{host_tmp.data(), size_, kOpFormat_NCHW, format_, in ConvertFormatAndSyncHostToDevice()
467 …const trans::FormatArgs format_args{host_ptr, size_, kOpFormat_NCHW, format_, host_shape, device_s… in ConvertFormatAndSyncHostToDevice()
|
| D | ge_types_convert.cc | 68 {kOpFormat_NCHW, ge::Format::FORMAT_NCHW}, in GetGeFormat()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter_util.cc | 143 …= TransformUtil::GetGeTensorDesc({static_cast<int>(vec.size())}, kNumberTypeInt32, kOpFormat_NCHW); in VectorToTensorUtil()
151 …= TransformUtil::GetGeTensorDesc({static_cast<int>(vec.size())}, kNumberTypeInt64, kOpFormat_NCHW); in VectorToTensorUtil()
159 …TransformUtil::GetGeTensorDesc({static_cast<int>(vec.size())}, kNumberTypeFloat32, kOpFormat_NCHW); in VectorToTensorUtil()
168 … = TransformUtil::GetGeTensorDesc({static_cast<int>(vec.size())}, kNumberTypeBool, kOpFormat_NCHW); in VectorToTensorUtil()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | single_tbe_json_creator.cc | 161 … def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_NCHW; in GenInputDescJson()
337 … def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_NCHW; in GenDescJson()
359 … def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_NCHW; in GenInputDescJson()
390 … def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_NCHW; in GenDescJson()
412 … def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_NCHW; in GenInputDescJson()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.cc | 266 tensor_format = kOpFormat_NCHW; in GenTopKV2IndicesTensorInfo()
268 std::string tensor_origin_format = kOpFormat_NCHW; in GenTopKV2IndicesTensorInfo()
375 return kOpFormat_NCHW; in FormatPass()
377 return origin_shape_size == kNCHWShapeSize ? kOpFormat_NCHW : kOpFormat_ND; in FormatPass()
|
| D | tbe_kernel_build.cc | 291 auto def_format = kOpFormat_NCHW; in GenValidInputDescJson()
514 auto def_format = kOpFormat_NCHW; in GenOutputList()
764 std::string format = kOpFormat_NCHW; in GetDeviceInputFormat()
773 format = kOpFormat_NCHW; in GetDeviceInputFormat()
806 std::string format = kOpFormat_NCHW; in GetDeviceOutputFormat()
815 format = kOpFormat_NCHW; in GetDeviceOutputFormat()
1098 auto def_format = kOpFormat_NCHW; in GenDescJson()
1131 format = ori_shape.size() == kNCHWShapeSize ? kOpFormat_NCHW : kOpFormat_ND; in GenDescJson()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | conv2d_grad_filter_gpu_kernel.h | 164 if (data_format_ == kOpFormat_NCHW || data_format_ == kOpFormat_DEFAULT) { in Init()
365 std::string data_format_ = kOpFormat_NCHW;
366 std::string format_attr_ = kOpFormat_NCHW;
|
| D | conv2d_grad_input_gpu_kernel.h | 138 if (format_attr == kOpFormat_NCHW) { in Init()
176 if (data_format_ == kOpFormat_NCHW || data_format_ == kOpFormat_DEFAULT) { in Init()
389 std::string data_format_ = kOpFormat_NCHW;
|
| D | conv2d_gpu_kernel.h | 134 if (data_format_ == kOpFormat_NCHW || data_format_ == kOpFormat_DEFAULT) { in Init()
368 std::string data_format_ = kOpFormat_NCHW;
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | graph_runner_test.cc | 68 conv2d->AddAttr("data_format", MakeValue(kOpFormat_NCHW)); in MakeGeGraph()
77 …d::vector<GeTensorPtr> ge_tensor_ptrs = TransformUtil::ConvertInputTensors(inputs, kOpFormat_NCHW); in DoExecGraph()
|
| D | convert_test.cc | 635 auto ge_tensor_ptr = TransformUtil::ConvertTensor(me_tensor_ptr, kOpFormat_NCHW); in TEST_F()
653 ASSERT_NE(TransformUtil::ConvertTensor(me_tensor_ptr, kOpFormat_NCHW), nullptr); in TEST_F()
676 ASSERT_EQ(TransformUtil::ConvertFormat(kOpFormat_NCHW), GeFormat::FORMAT_NCHW); in TEST_F()
770 …td::vector<GeTensorPtr> ge_tensors = TransformUtil::ConvertInputTensors(me_inputs, kOpFormat_NCHW); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pynative/ |
| D | pynative_execute_test.cc | 62 … "stride"_a = 1, "dilation"_a = 1, "group"_a = 1, "data_format"_a = kOpFormat_NCHW); in ConstructOpExecInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel.h | 158 if (((origin_data_format == kOpFormat_DEFAULT) || (origin_data_format == kOpFormat_NCHW)) && in AxisTransform()
161 } else if (((cal_format == kOpFormat_DEFAULT) || (cal_format == kOpFormat_NCHW)) && in AxisTransform()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute_ge.cc | 100 …ansformUtil::GetGeTensorDesc(me_tensor_ptr->shape_c(), me_tensor_ptr->data_type(), kOpFormat_NCHW); in SetInputsForSingleOpGraph()
181 auto ge_tensor_ptr = transform::TransformUtil::ConvertTensor(me_tensor_ptr, kOpFormat_NCHW); in ToTensorPtr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.h | 128 const std::set<std::string> kCommonFormatSet = {kOpFormat_DEFAULT, kOpFormat_ND, kOpFormat_NCHW, kO…
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 574 constexpr auto kOpFormat_NCHW = "NCHW"; variable
597 kOpFormat_NCHW,
616 const std::set<std::string> kDefaultCompatibleFormat = {kOpFormat_ND, kOpFormat_NCHW, kOpFormat_NHW…
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/ |
| D | op_node.cc | 260 …>format == kOpFormat_DEFAULT || input->format == kOpFormat_NHWC || input->format == kOpFormat_NCHW; in InferShape()
266 input->format == kOpFormat_NCHW || input->format == kOpFormat_FRAC_NZ; in InferShape()
453 if (ori_format == kOpFormat_DEFAULT || ori_format == kOpFormat_NCHW) { in InferFormat()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | transform_op_optimizer.cc | 214 {{kOpFormat_NCHW, kOpFormat_NHWC}, {0, 2, 3, 1}}, in GenTransformOp()
215 {{kOpFormat_NHWC, kOpFormat_NCHW}, {0, 3, 1, 2}}, in GenTransformOp()
|