Home
last modified time | relevance | path

Searched refs:kOpFormat_HWCN (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dtransdata_split.cc78 RefreshKernelBuildInfo(input_format, kOpFormat_HWCN, new_transdata_node, padding_axis); in DoSplit()
82 RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, new_transpose_node); in DoSplit()
91 RefreshKernelBuildInfo(input_format, kOpFormat_HWCN, new_transpose_node); in DoSplit()
96 RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, new_transdata_node, padding_axis); in DoSplit()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_device_address.cc59 …std::make_pair("float16", mindspore::kOpFormat_HWCN), std::make_pair("float32", mindspore::kOpF…
60 …std::make_pair("int8", mindspore::kOpFormat_HWCN), std::make_pair("int16", mindspore::kOpFor…
61 …std::make_pair("int32", mindspore::kOpFormat_HWCN), std::make_pair("int64", mindspore::kOpFor…
62 …std::make_pair("uint8", mindspore::kOpFormat_HWCN), std::make_pair("uint16", mindspore::kOpFo…
63 …std::make_pair("uint32", mindspore::kOpFormat_HWCN), std::make_pair("uint64", mindspore::kOpFo…
72 …kOpFormat_NHWC, kOpFormat_HWCN, kOpFormat_NC1HWC0, kOpFormat_FRAC_Z, kOpFormat_C…
Dge_types_convert.cc70 {kOpFormat_HWCN, ge::Format::FORMAT_HWCN}, in GetGeFormat()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_transpose_for_dynamic_gru_v2.cc63 RefreshKernelBuildInfo(input_format, kOpFormat_HWCN, new_transpose_node); in Insert()
68 RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, new_transdata_node, padding_axis); in Insert()
/third_party/mindspore/tests/ut/cpp/common/
Dtrans_test.cc39 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()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h576 constexpr auto kOpFormat_HWCN = "HWCN"; variable
599 kOpFormat_HWCN,
616 …::string> kDefaultCompatibleFormat = {kOpFormat_ND, kOpFormat_NCHW, kOpFormat_NHWC, kOpFormat_HWCN,
/third_party/mindspore/mindspore/ccsrc/common/
Dtrans.cc976 … {kOpFormat_HWCN, HwchDeviceShape}, in TransShapeToDevice()
1023 {kOpFormat_HWCN, HwchDeviceDynamicShape}, in TransShapeToDevice()
1109 if (args.device_format == kOpFormat_HWCN || args.device_format == kOpFormat_NHWC) { in TransFormat()
1142 if (args.device_format == kOpFormat_HWCN || args.device_format == kOpFormat_NHWC) { in TransFormatFromDeviceToHost()
1185 } else if (args.device_format == kOpFormat_HWCN) { in NchwTo4D()
1217 } else if (args.device_format == kOpFormat_HWCN) { in ToNchw()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dutil.cc96 } else if (format == kOpFormat_HWCN) { in ConvertFormat()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_dynaminc_shape_util.cc262 {kOpFormat_HWCN, HwchRange}, in DynamicShapeRangeTrans()