Searched defs:output_grad_shape (Results 1 – 4 of 4) sorted by relevance
242 Output output_grad_shape = Shape(scope, output_grad); in EinsumGradReducedHelper() local
630 TensorShape output_grad_shape = ctx->InputShape("grad_output"); in Compile() local
1461 const Shape& output_grad_shape, int64_t feature_index) { in InferBatchNormGradShape()
898 ShapeHandle output_grad_shape = c->input(2); in Conv2DBackpropInputShape() local