Searched refs:InferBatchNormGradShape (Results 1 – 4 of 4) sorted by relevance
101 static StatusOr<Shape> InferBatchNormGradShape(const Shape& operand_shape,
744 return CheckShape(batch_norm_grad, ShapeInference::InferBatchNormGradShape( in HandleBatchNormGrad()
1415 /* static */ StatusOr<Shape> ShapeInference::InferBatchNormGradShape( in InferBatchNormGradShape() function in xla::ShapeInference
2107 ShapeInference::InferBatchNormGradShape( in BatchNormGrad()