Searched refs:kAttrKeepDims (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | confusion_softmax_grad_rule.cc | 44 if (!GetBoolAttr(sum_anf, kAttrKeepDims)) { in Process()
57 AnfAlgo::CopyNodeAttr(kAttrKeepDims, sum_anf, fusion_node); in Process()
|
| D | softmax_grad_ext_fusion.cc | 63 if (!GetBoolAttr(sum, kAttrKeepDims)) { in Process()
73 AnfAlgo::CopyNodeAttr(kAttrKeepDims, "keepdims", sum, fusion_node); in Process()
|
| D | square_sum_fusion.cc | 52 AnfAlgo::CopyNodeAttr(kAttrKeepDims, sum, square_sumv1); in GenerateSquareSumV1()
73 AnfAlgo::CopyNodeAttr(kAttrKeepDims, sum, square_sumv2); in GenerateSquareSumV2()
|
| D | confusion_mul_grad_fusion.cc | 48 AnfAlgo::CopyNodeAttr(kAttrKeepDims, reduce_sum, fusion_node); in CreateFusionNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | reduce_min_fission.cc | 32 AnfAlgo::CopyNodeAttr(kAttrKeepDims, old_node, reduce_min); in CreateReduceMin()
119 if (!prim->HasAttr(kAttrAxis) || !prim->HasAttr(kAttrKeepDims)) { in Process()
129 auto keep_dims = AnfAlgo::GetNodeAttr<bool>(cnode, kAttrKeepDims); in Process()
|
| D | dynamic_rnn_grad_fission_v2.cc | 463 AnfAlgo::SetNodeAttr(kAttrKeepDims, MakeValue(false), reduce_sum); in CreateDwReduceSum()
513 AnfAlgo::SetNodeAttr(kAttrKeepDims, MakeValue(false), reduce_sum); in CreateDbReduceSum()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_reduce_selecter.cc | 136 if (!AnfAlgo::HasNodeAttr(kAttrKeepDims, cnode_ptr_)) { in GetReduceAttrKeepDim()
141 keep_dims_ = AnfAlgo::GetNodeAttr<bool>(cnode_ptr_, kAttrKeepDims); in GetReduceAttrKeepDim()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 354 constexpr auto kAttrKeepDims = "keep_dims"; variable
|