Home
last modified time | relevance | path

Searched refs:kBNTrainingUpdateV2OutputNum (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbatch_norm_bert_fission.cc157 CreateMultipleOutputsOfAnfNode(func_graph, bn_training_update_v2, kBNTrainingUpdateV2OutputNum, in Process()
159 if (bn_training_update_v2_outputs.size() != kBNTrainingUpdateV2OutputNum) { in Process()
160 …EXCEPTION) << "The output size of node bn_training_reduce must be " << kBNTrainingUpdateV2OutputNum in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.h71 constexpr size_t kBNTrainingUpdateV2OutputNum = 3; variable