Searched refs:InferBatchNormTrainingShape (Results 1 – 4 of 4) sorted by relevance
88 static StatusOr<Shape> InferBatchNormTrainingShape(const Shape& operand_shape,
724 ShapeInference::InferBatchNormTrainingShape( in HandleBatchNormTraining()
1166 /* static */ StatusOr<Shape> ShapeInference::InferBatchNormTrainingShape( in InferBatchNormTrainingShape() function in xla::ShapeInference
2056 ShapeInference::InferBatchNormTrainingShape( in BatchNormTraining()