Home
last modified time | relevance | path

Searched refs:kOpFormat_NC1HWC0 (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbn_grad_split_test.cc76 builder2.SetOutputsFormat({kOpFormat_NC1HWC0}); in TEST_F()
83 {kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, in TEST_F()
84 kOpFormat_NC1HWC0}); in TEST_F()
86 {kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, in TEST_F()
87 kOpFormat_NC1HWC0}); in TEST_F()
142 builder2.SetOutputsFormat({kOpFormat_NC1HWC0}); in TEST_F()
149 … {kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0}); in TEST_F()
151 … {kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0}); in TEST_F()
Dbn_split_test.cc79 … {kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0}); in TEST_F()
81 … {kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0}); in TEST_F()
128 … {kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0}); in TEST_F()
130 … {kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0, kOpFormat_NC1HWC0}); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dmerge_cast_to_op_test.cc52 builder.SetOutputsFormat({kOpFormat_NC1HWC0}); in Query()
58 builder.SetInputsFormat({kOpFormat_NC1HWC0}); in Query()
98 builder1.SetOutputsFormat({kOpFormat_NC1HWC0}); in TEST_F()
173 builder2.SetInputsFormat({kOpFormat_NC1HWC0}); in TEST_F()
Dremove_internal_output_test.cc58 builder.SetOutputsFormat({kOpFormat_NC1HWC0}); in GetSingleOutputGraph()
87 builder.SetOutputsFormat({kOpFormat_NC1HWC0, kOpFormat_NC1HWC0}); in GetMutilpleOutputGraph()
102 builder.SetInputsFormat({kOpFormat_NC1HWC0}); in SelectKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_broadcast_selecter.cc73 AssignSupportFormat(kOpFormat_NC1HWC0, support_format); in IsBroadCastSupport5HD()
92 input_support_format.emplace_back(kOpFormat_NC1HWC0); in IsBroadCastSupport5HD()
109 input_support_format.assign(input_num_, kOpFormat_NC1HWC0); in IsBroadCastSupport5HD()
111 GenOutputSupportFormat(kOpFormat_NC1HWC0, &output_support_format); in IsBroadCastSupport5HD()
Dtbe_kernel_reduce_selecter.cc69 AssignSupportFormat(kOpFormat_NC1HWC0, support_format); in IsReduceSupport5HD()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dspace_to_depth_split.cc45 tensor::DeviceInfo device_info{kOpFormat_NC1HWC0, tensor_type}; in CreateTensor()
84 op_builder.SetOutputsFormat({kOpFormat_NC1HWC0}); in CreateValueNode()
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc169 CreateKernelBuildInfo({kOpFormat_DEFAULT, kOpFormat_NC1HWC0}, {kOpFormat_DEFAULT}, in TEST_F()
180 CreateKernelBuildInfo({kOpFormat_DEFAULT, kOpFormat_NC1HWC0}, {kOpFormat_DEFAULT}, in TEST_F()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_device_address.cc50 …std::make_pair("float16", mindspore::kOpFormat_NC1HWC0), std::make_pair("float32", mindspore::kOpF…
51 …std::make_pair("bool", mindspore::kOpFormat_NC1HWC0), std::make_pair("float32", mindspore::kOpF…
72 …kOpFormat_NHWC, kOpFormat_HWCN, kOpFormat_NC1HWC0, kOpFormat_FRAC_Z, kOpFormat_C…
Dge_types_convert.cc71 {kOpFormat_NC1HWC0, ge::Format::FORMAT_NC1HWC0}, in GetGeFormat()
Dkernel_select_ascend.cc56 {prim::kPrimConv2D->name(), {kOpFormat_NC1HWC0, kOpFormat_FRAC_Z}}};
79 string priority_matched_format = kOpFormat_NC1HWC0; in GetPriorityMatchFormat()
Dkernel_select_graph_kernel.cc218 *default_format = kOpFormat_NC1HWC0; in GetDefaultFormat()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h577 constexpr auto kOpFormat_NC1HWC0 = "NC1HWC0"; variable
600 kOpFormat_NC1HWC0,
668 kOpFormat_FRACTAL_Z_3D, kOpFormat_NC1KHKWHWC0, kOpFormat_NC1HWC0, kOpFormat_FRAC_NZ,
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc257 builder.SetOutputsFormat({kOpFormat_NCHW, kOpFormat_NC1HWC0}); in TEST_F()
260 EXPECT_EQ(AnfAlgo::GetOutputFormat(add, 1), kOpFormat_NC1HWC0); in TEST_F()
276 builder.SetInputsFormat({kOpFormat_NCHW, kOpFormat_NC1HWC0}); in TEST_F()
279 EXPECT_EQ(AnfAlgo::GetInputFormat(add, 1), kOpFormat_NC1HWC0); in TEST_F()
479 builder.SetInputsFormat({kOpFormat_NCHW, kOpFormat_NC1HWC0}); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Dmemcpy_async.cc170kOpFormat_NC1HWC0, kOpFormat_FRAC_Z, kOpFormat_NC1KHKWHWC0,
/third_party/mindspore/mindspore/ccsrc/common/
Dtrans.h133 {kOpFormat_NC1HWC0, NchwToNc1hwc0}, {kOpFormat_C1HWNCoC0, NchwToC1hwncoc0},
Dtrans.cc978 … {kOpFormat_NC1HWC0, Nc1hwc0DeviceShape}, in TransShapeToDevice()
1025 {kOpFormat_NC1HWC0, Nc1hwc0DeviceDynamicShape}, in TransShapeToDevice()
1133 {kOpFormat_NC1HWC0, Nc1hwc0ToNchw}, {kOpFormat_C1HWNCoC0, C1hwncoc0ToNchw}, in TransFormatFromDeviceToHost()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dutil.cc92 } else if (format == kOpFormat_NC1HWC0) { in ConvertFormat()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_dynaminc_shape_util.cc263 {kOpFormat_NC1HWC0, Nc1hwc0Range}, in DynamicShapeRangeTrans()
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc677 ASSERT_EQ(TransformUtil::ConvertFormat(kOpFormat_NC1HWC0), GeFormat::FORMAT_NC1HWC0); in TEST_F()