Searched refs:output_shape_for_mean_and_var (Results 1 – 1 of 1) sorted by relevance
1247 Shape output_shape_for_mean_and_var = in InferBatchNormTrainingShape() local1267 output_shape_for_mean_and_var, in InferBatchNormTrainingShape()1268 output_shape_for_mean_and_var}); in InferBatchNormTrainingShape()1377 Shape output_shape_for_mean_and_var = in InferBatchNormInferenceShape() local