Searched refs:kBN1OutputNum (Results 1 – 2 of 2) sorted by relevance
119 if (bn_training_reduce_outputs.size() != kBN1OutputNum) { in SplitBatchNormForTBE()144 if (bn_training_reduce_outputs.size() != kBN1OutputNum) { in SyncBNSplitForTBE()
58 constexpr size_t kBN1OutputNum = 2; variable