Searched refs:DynamicElementCountPerFeature (Results 1 – 1 of 1) sorted by relevance
112 std::unique_ptr<HloInstruction> DynamicElementCountPerFeature( in DynamicElementCountPerFeature() function in xla::__anon262de65b0111::BatchNormExpanderVisitor232 add(DynamicElementCountPerFeature(operand, feature_index, add)); in HandleBatchNormTraining()473 add(DynamicElementCountPerFeature(activation, feature_index, add)); in HandleBatchNormGrad()