Searched refs:FusedBatchNormShape (Results 1 – 3 of 3) sorted by relevance
252 Status FusedBatchNormShape(shape_inference::InferenceContext* c);
789 Status FusedBatchNormShape(shape_inference::InferenceContext* c) { in FusedBatchNormShape() function
183 .SetShapeFn(shape_inference::FusedBatchNormShape);201 .SetShapeFn(shape_inference::FusedBatchNormShape);