Searched refs:specified_input_grad_shape (Results 1 – 1 of 1) sorted by relevance
877 ShapeHandle specified_input_grad_shape; in Conv2DBackpropInputShape() local879 c->MakeShapeFromShapeTensor(0, &specified_input_grad_shape)); in Conv2DBackpropInputShape()880 if (c->Rank(specified_input_grad_shape) == InferenceContext::kUnknownRank) { in Conv2DBackpropInputShape()881 TF_RETURN_IF_ERROR(c->WithRank(specified_input_grad_shape, 4, in Conv2DBackpropInputShape()882 &specified_input_grad_shape)); in Conv2DBackpropInputShape()891 int specified_input_grad_rank = c->Rank(specified_input_grad_shape); in Conv2DBackpropInputShape()895 specified_input_grad_shape, data_format, &specified_batch_size_dim, in Conv2DBackpropInputShape()902 c->Dim(specified_input_grad_shape, 0); in Conv2DBackpropInputShape()904 c->Dim(specified_input_grad_shape, 1); in Conv2DBackpropInputShape()