Searched refs:kBnInputTensorNum (Results 1 – 4 of 4) sorted by relevance
42 if (AnfAlgo::GetInputTensorNum(bn_cnode) != kBnInputTensorNum) { in CreateOutputsOfBNTrainingReduce()43 …MS_LOG(INFO) << "BatchNorm's input size less than " << kBnInputTensorNum << ". " << bn_cnode->Debu… in CreateOutputsOfBNTrainingReduce()74 CheckCNodeInputSize(bn_cnode, kBnInputTensorNum); in CreateOutputsOfBNTrainingUpdate()109 if (AnfAlgo::GetInputTensorNum(cnode) < kBnInputTensorNum) { in SplitBatchNormForTBE()110 …MS_LOG(INFO) << "op[" << cnode->DebugString() << "] has less input than " << kBnInputTensorNum << … in SplitBatchNormForTBE()134 if (AnfAlgo::GetInputTensorNum(cnode) < kBnInputTensorNum) { in SyncBNSplitForTBE()135 …MS_LOG(INFO) << "op[" << cnode->DebugString() << "] has less input than " << kBnInputTensorNum << … in SyncBNSplitForTBE()
65 if (AnfAlgo::GetInputTensorNum(batchnorm) < kBnInputTensorNum) { in CheckBatchNorm()66 MS_LOG(DEBUG) << "BatchNorm's input number less than " << kBnInputTensorNum; in CheckBatchNorm()
68 MS_LOG(DEBUG) << "BatchNormGrad's input number less than " << kBnInputTensorNum; in CheckBatchNormGrad()
55 constexpr size_t kBnInputTensorNum = 5; variable