Searched defs:output_grad_shape (Results 1 – 3 of 3) sorted by relevance
632 TensorShape output_grad_shape = ctx->InputShape("grad_output"); in Compile() local
1451 const Shape& output_grad_shape, int64_t feature_index) { in InferBatchNormGradShape()
883 ShapeHandle output_grad_shape = c->input(2); in Conv2DBackpropInputShape() local