Searched refs:FusedBatchNormGradShape (Results 1 – 3 of 3) sorted by relevance
255 Status FusedBatchNormGradShape(shape_inference::InferenceContext* c);
824 Status FusedBatchNormGradShape(shape_inference::InferenceContext* c) { in FusedBatchNormGradShape() function
218 .SetShapeFn(shape_inference::FusedBatchNormGradShape);236 .SetShapeFn(shape_inference::FusedBatchNormGradShape);