Searched refs:output_grad_depth_dim (Results 1 – 1 of 1) sorted by relevance
904 DimensionHandle output_grad_depth_dim; in Conv2DBackpropInputShape() local908 absl::MakeSpan(output_grad_spatial_dims), &output_grad_depth_dim, c)); in Conv2DBackpropInputShape()911 c->Merge(output_grad_depth_dim, c->Dim(filter_shape, 3), &unused)); in Conv2DBackpropInputShape()