Searched refs:specified_input_grad_spatial_dims (Results 1 – 1 of 1) sorted by relevance
890 gtl::InlinedVector<DimensionHandle, 2> specified_input_grad_spatial_dims(2); in Conv2DBackpropInputShape() local896 absl::MakeSpan(specified_input_grad_spatial_dims), in Conv2DBackpropInputShape()901 specified_input_grad_spatial_dims[0] = in Conv2DBackpropInputShape()903 specified_input_grad_spatial_dims[1] = in Conv2DBackpropInputShape()915 batch_size_dim, specified_input_grad_spatial_dims, input_grad_depth_dim, in Conv2DBackpropInputShape()