Searched refs:kOpFormat_NDC1HWC0 (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_broadcast_selecter.cc | 258 AssignSupportFormat(kOpFormat_NDC1HWC0, support_format); in IsBroadCastSupportNDC1HWC0()
274 input_support_format.emplace_back(kOpFormat_NDC1HWC0); in IsBroadCastSupportNDC1HWC0()
291 input_support_format.assign(input_num_, kOpFormat_NDC1HWC0); in IsBroadCastSupportNDC1HWC0()
293 GenOutputSupportFormat(kOpFormat_NDC1HWC0, &output_support_format); in IsBroadCastSupportNDC1HWC0()
|
| D | tbe_kernel_reduce_selecter.cc | 85 AssignSupportFormat(kOpFormat_NDC1HWC0, support_format); in IsReduceSupportNDC1HWC0()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | max_pool3d_grad_grad_fission.cc | 51 tensor::DeviceInfo device_info{kOpFormat_NDC1HWC0, tensor_type}; in CreateTensor()
87 op_builder.SetOutputsFormat({kOpFormat_NDC1HWC0}); in CreateValueNode()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 588 constexpr auto kOpFormat_NDC1HWC0 = "NDC1HWC0"; variable
610 kOpFormat_NDC1HWC0,
670 kOpFormat_FRACTAL_ZN_RNN, kOpFormat_NDC1HWC0, kOpFormat_FRAC_Z};
678 const std::set<std::string> k3DFormatSet = {kOpFormat_NCDHW, kOpFormat_NDC1HWC0, kOpFormat_FRACTAL_…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ge_types_convert.cc | 81 {kOpFormat_NDC1HWC0, ge::Format::FORMAT_NDC1HWC0}, in GetGeFormat()
|
| D | kernel_select_ascend.cc | 105 priority_matched_format = kOpFormat_NDC1HWC0; in GetPriorityMatchFormat()
|
| D | ascend_device_address.cc | 73 …kOpFormat_FRAC_NZ, kOpFormat_NC1HWC0_C04, kOpFormat_FRACTAL_Z_C04, kOpFormat_NDC1HWC0, kOpFormat_F…
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.h | 135 {kOpFormat_NDC1HWC0, NcdhwToNdc1hwc0}, {kOpFormat_FRACTAL_Z_3D, NcdhwToFracZ3D}};
|
| D | trans.cc | 984 … {kOpFormat_NDC1HWC0, Ndc1hwc0DeviceShape}, in TransShapeToDevice()
1031 {kOpFormat_NDC1HWC0, Ndc1hwc0DeviceDynamicShape}, in TransShapeToDevice()
1134 {kOpFormat_NC1HWC0_C04, Nc1hwc04ToNchw}, {kOpFormat_NDC1HWC0, Ndc1hwc0ToNcdhw}, in TransFormatFromDeviceToHost()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_dynaminc_shape_util.cc | 270 {kOpFormat_NDC1HWC0, Ndc1hwc0Range}, in DynamicShapeRangeTrans()
|