Searched refs:variance_shape (Results 1 – 3 of 3) sorted by relevance
98 const Shape& variance_shape, int64 feature_index);
1318 const Shape& variance_shape, int64 feature_index) { in InferBatchNormInferenceShape() argument1331 ShapeUtil::ValidateShapeWithOptionalLayout(variance_shape)); in InferBatchNormInferenceShape()1409 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(variance_shape, in InferBatchNormInferenceShape()1417 PrimitiveType_Name(variance_shape.element_type())); in InferBatchNormInferenceShape()1448 if (ShapeUtil::GetDimension(variance_shape, 0) != feature_count) { in InferBatchNormInferenceShape()1453 ShapeUtil::GetDimension(variance_shape, 0), feature_count); in InferBatchNormInferenceShape()
2673 TF_ASSIGN_OR_RETURN(const Shape* variance_shape, GetShapePtr(variance)); in BatchNormInference()2677 *mean_shape, *variance_shape, feature_index)); in BatchNormInference()