Searched refs:output_shape_for_mean_and_var (Results 1 – 1 of 1) sorted by relevance
1291 Shape output_shape_for_mean_and_var = in InferBatchNormTrainingShape() local1311 output_shape_for_mean_and_var, in InferBatchNormTrainingShape()1312 output_shape_for_mean_and_var}); in InferBatchNormTrainingShape()1421 Shape output_shape_for_mean_and_var = in InferBatchNormInferenceShape() local