Searched refs:kAttrFormat (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_utils.cc | 75 return anf_node->isa<CNode>() && AnfAlgo::HasNodeAttr(kAttrFormat, anf_node->cast<CNodePtr>()) && in IsNeedChangeDefaultFormat()
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/optimizer/ascend/ir_fusion/ |
| D | bn_reduce_grad_conv2d_backprop_filter_fusion.cc | 105 AnfAlgo::CopyNodeAttr(kAttrFormat, conv_back_filter, fused_dbn_dw); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | conv2d_unify_mindir.cc | 54 auto data_format_ptr = primitive_ptr->GetAttr(kAttrFormat); in NeedUpdate()
215 AnfAlgo::CopyNodeAttr(kAttrFormat, conv2d, depth_conv); in SetCommonAttrs()
|
| D | avg_pool_grad_unify_mindir.cc | 213 AnfAlgo::CopyNodeAttr(kAttrFormat, avgpool_grad, avgpool_grad_vm); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 350 constexpr auto kAttrFormat = "format"; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.cc | 117 if (AnfAlgo::HasNodeAttr(kAttrFormat, cnode->cast<CNodePtr>())) { in IsNeedChangeDefaultFormat()
118 auto attr = AnfAlgo::GetNodeAttr<std::string>(cnode, kAttrFormat); in IsNeedChangeDefaultFormat()
|