Searched refs:GetBoolAttr (Results 1 – 14 of 14) 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()
|
| D | softmax_grad_ext_fusion.cc | 63 if (!GetBoolAttr(sum, kAttrKeepDims)) { in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | cdist_fission.cc | 100 if (GetBoolAttr(cdist_cnode, kAttrVisited)) { in Process()
130 if (GetBoolAttr(cdist_grad_cnode, kAttrVisited)) { in Process()
|
| D | bce_with_logits_loss_fission.cc | 86 if (GetBoolAttr(cnode, kAttrVisited)) { in Process()
|
| D | single_batch_norm_fission.cc | 106 if (!GetBoolAttr(cnode, kAttrIsTraining)) { in Process()
|
| D | bn_grad_split.cc | 149 if (!GetBoolAttr(cnode, kAttrIsTraining)) { in Process()
|
| D | bn_split.cc | 258 if (!GetBoolAttr(node, kAttrIsTraining)) { in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | bce_with_logits_loss_fusion.cc | 79 if (GetBoolAttr(cnode, kAttrVisited)) { in Process()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | resizebilinear_info.cc | 44 align_corners_ = GetBoolAttr(ALIGN_CORNERS); in GetAttrs()
|
| D | batchnorm_info.cc | 33 is_training_ = GetBoolAttr(IS_TRAINING); in GetAttrs()
|
| D | operator_info.h | 222 bool GetBoolAttr(const std::string &attr_name);
|
| D | operator_info.cc | 1781 bool OperatorInfo::GetBoolAttr(const std::string &attr_name) { in GetBoolAttr() function in mindspore::parallel::OperatorInfo
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.h | 211 bool GetBoolAttr(const AnfNodePtr &node, const std::string &attr_name);
|
| D | helper.cc | 874 bool GetBoolAttr(const AnfNodePtr &node, const std::string &attr_name) { in GetBoolAttr() function
|