| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | trans_op_format_refine.cc | 42 builder->SetInputsFormat({kOpFormat_NCDHW}); in Process()
45 AnfAlgo::SetNodeAttr(kAttrSrcFormat, MakeValue(kOpFormat_NCDHW), node); in Process()
49 builder->SetOutputsFormat({kOpFormat_NCDHW}); in Process()
51 AnfAlgo::SetNodeAttr(kAttrDstFormat, MakeValue(kOpFormat_NCDHW), node); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | single_tbe_json_creator.cc | 161 …auto def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_… in GenInputDescJson()
165 …(def_format == kOpFormat_NCDHW && k3DFormatSet.find(format) == k3DFormatSet.end()) ? kOpFormat_NCD… in GenInputDescJson()
337 …auto def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_… in GenDescJson()
359 …auto def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_… in GenInputDescJson()
390 …auto def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_… in GenDescJson()
412 …auto def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_… in GenInputDescJson()
|
| D | tbe_json_creator.cc | 414 …auto def_format = TbeJsonUtils::IsNeedChangeDefaultFormat(anf_node) ? kOpFormat_NCDHW : kOpFormat_… in GenDescJson()
416 …(def_format == kOpFormat_NCDHW && k3DFormatSet.find(format) == k3DFormatSet.end()) ? kOpFormat_NCD… in GenDescJson()
|
| D | tbe_json_utils.cc | 76 AnfAlgo::GetNodeAttr<std::string>(anf_node, kAttrFormat) == kOpFormat_NCDHW; in IsNeedChangeDefaultFormat()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | add_attr_for_3d_graph.cc | 36 AnfAlgo::SetNodeAttr(kAttrFormat, MakeValue(kOpFormat_NCDHW), node); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | conv3d_transpose_gpu_kernel.h | 65 if (data_format_ == kOpFormat_NCDHW || data_format_ == kOpFormat_DEFAULT) { in Launch()
132 if (format_attr == kOpFormat_NCDHW) { in Init()
190 if (data_format_ == kOpFormat_NCDHW || data_format_ == kOpFormat_DEFAULT) { in Init()
487 if (data_format_ == kOpFormat_NCDHW || data_format_ == kOpFormat_DEFAULT) { in UsePadProcess()
528 std::string data_format_ = kOpFormat_NCDHW;
|
| D | conv3d_gpu_kernel.h | 90 data_format_ = kOpFormat_NCDHW; in Init()
139 if (data_format_ != kOpFormat_NCDHW) { in Init()
365 std::string data_format_ = kOpFormat_NCDHW;
|
| D | conv3d_grad_input_gpu_kernel.h | 90 data_format_ = kOpFormat_NCDHW; in Init()
140 if (data_format_ != kOpFormat_NCDHW) { in Init()
374 std::string data_format_ = kOpFormat_NCDHW;
|
| D | conv3d_grad_filter_gpu_kernel.h | 117 data_format_ = kOpFormat_NCDHW; in Init()
163 if (data_format_ != kOpFormat_NCDHW) { in Init()
399 std::string data_format_ = kOpFormat_NCDHW;
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 585 constexpr auto kOpFormat_NCDHW = "NCDHW"; variable
611 kOpFormat_NCDHW,
617 kOpFormat_NCDHW};
678 const std::set<std::string> k3DFormatSet = {kOpFormat_NCDHW, kOpFormat_NDC1HWC0, kOpFormat_FRACTAL_…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.cc | 119 return attr == kOpFormat_NCDHW; in IsNeedChangeDefaultFormat()
297 def_format = kOpFormat_NCDHW; in GenValidInputDescJson()
299 if (def_format == kOpFormat_NCDHW && k3DFormatSet.find(format) == k3DFormatSet.end()) { in GenValidInputDescJson()
300 format = kOpFormat_NCDHW; in GenValidInputDescJson()
516 def_format = kOpFormat_NCDHW; in GenOutputList()
525 if (def_format == kOpFormat_NCDHW && k3DFormatSet.find(format) == k3DFormatSet.end()) { in GenOutputList()
526 format = kOpFormat_NCDHW; in GenOutputList()
766 format = kOpFormat_NCDHW; in GetDeviceInputFormat()
808 format = kOpFormat_NCDHW; in GetDeviceOutputFormat()
1100 def_format = kOpFormat_NCDHW; in GenDescJson()
|
| D | tbe_dynaminc_shape_util.cc | 268 {kOpFormat_NCDHW, NcdhwRange}, in DynamicShapeRangeTrans()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.h | 128 …std::string> kCommonFormatSet = {kOpFormat_DEFAULT, kOpFormat_ND, kOpFormat_NCHW, kOpFormat_NCDHW};
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | max_pool3d_grad_grad_fission.cc | 37 if (data_format != kOpFormat_NCDHW) { in CreateTensor()
|
| /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()
245 if (format_ == kOpFormat_FRAC_NZ || format_ == kOpFormat_NCDHW) { in GetDeviceShape()
375 …if (format_ == kOpFormat_NCHW || format_ == kOpFormat_DEFAULT || format_ == kOpFormat_NCDHW || for… in SyncHostToDevice()
|
| D | ge_types_convert.cc | 78 {kOpFormat_NCDHW, ge::Format::FORMAT_NCDHW}, in GetGeFormat()
|
| D | kernel_select_ascend.cc | 168 kernel_build_info.GetInputFormat(input_index) == kOpFormat_NCDHW) { in UpdateCurMatchCounts()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_broadcast_selecter.cc | 268 input_support_format.emplace_back(kOpFormat_NCDHW); in IsBroadCastSupportNDC1HWC0()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | util.cc | 86 } else if (format == kOpFormat_NCDHW) { in ConvertFormat()
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 982 … {kOpFormat_NCDHW, NcdhwDeviceShape}, in TransShapeToDevice()
1029 {kOpFormat_NCDHW, NcdhwDeviceDynamicShape}, in TransShapeToDevice()
|