Searched defs:output_grad_shape (Results 1 – 2 of 2) sorted by relevance
632 TensorShape output_grad_shape = ctx->InputShape("grad_output"); in Compile() local
1418 const Shape& output_grad_shape, int64 feature_index) { in InferBatchNormGradShape()