Searched refs:variance_shape (Results 1 – 3 of 3) sorted by relevance
98 const Shape& variance_shape, int64 feature_index);
1274 const Shape& variance_shape, int64 feature_index) { in InferBatchNormInferenceShape() argument1287 ShapeUtil::ValidateShapeWithOptionalLayout(variance_shape)); in InferBatchNormInferenceShape()1365 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(variance_shape, in InferBatchNormInferenceShape()1373 PrimitiveType_Name(variance_shape.element_type())); in InferBatchNormInferenceShape()1404 if (ShapeUtil::GetDimension(variance_shape, 0) != feature_count) { in InferBatchNormInferenceShape()1409 ShapeUtil::GetDimension(variance_shape, 0), feature_count); in InferBatchNormInferenceShape()
2079 TF_ASSIGN_OR_RETURN(const Shape& variance_shape, GetShape(variance)); in BatchNormInference()2083 variance_shape, feature_index)); in BatchNormInference()