Searched refs:variance_shape (Results 1 – 3 of 3) sorted by relevance
96 const Shape& variance_shape, int64_t feature_index);
1317 const Shape& variance_shape, int64_t feature_index) { in InferBatchNormInferenceShape() argument1330 ShapeUtil::ValidateShapeWithOptionalLayout(variance_shape)); in InferBatchNormInferenceShape()1408 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(variance_shape, in InferBatchNormInferenceShape()1416 PrimitiveType_Name(variance_shape.element_type())); in InferBatchNormInferenceShape()1447 if (ShapeUtil::GetDimension(variance_shape, 0) != feature_count) { in InferBatchNormInferenceShape()1452 ShapeUtil::GetDimension(variance_shape, 0), feature_count); in InferBatchNormInferenceShape()
2827 TF_ASSIGN_OR_RETURN(const Shape* variance_shape, GetShapePtr(variance)); in BatchNormInference()2831 *mean_shape, *variance_shape, feature_index)); in BatchNormInference()