Searched refs:output_spatial_dims (Results 1 – 3 of 3) sorted by relevance
189 auto output_spatial_dims = conv_dnums.output_spatial_dimensions(); in MatchBackwardFilter() local229 conv->shape().dimensions(output_spatial_dims[0])) && in MatchBackwardFilter()245 int64 filter_size = conv->shape().dimensions(output_spatial_dims[i]); in MatchBackwardFilter()325 for (int i = 0; i < output_spatial_dims.size(); ++i) { in MatchBackwardFilter()326 backward_conv_dnums.add_kernel_spatial_dimensions(output_spatial_dims[i]); in MatchBackwardFilter()399 const auto& output_spatial_dims = dnums.output_spatial_dimensions(); in MatchBackwardInput() local401 CHECK_EQ(output_spatial_dims.size(), input_spatial_dims.size()); in MatchBackwardInput()439 auto unpadded_input_size = conv->shape().dimensions(output_spatial_dims[i]); in MatchBackwardInput()
112 llvm::SmallVector<int64_t, 4> output_spatial_dims( in ConvertConvDimensionNumbers() local124 Convert(output_spatial_dims, builder), builder->getContext()); in ConvertConvDimensionNumbers()
2436 std::vector<int64_t> output_spatial_dims = in ConvertConv2DHelper() local2438 const int output_height = output_spatial_dims[0]; in ConvertConv2DHelper()2439 const int output_width = output_spatial_dims[1]; in ConvertConv2DHelper()