Searched refs:kernel_spatial_dims (Results 1 – 3 of 3) sorted by relevance
123 auto kernel_spatial_dims = conv_dnums.kernel_spatial_dimensions(); in MatchBackwardFilter() local237 for (int i = 0; i < kernel_spatial_dims.size(); ++i) { in MatchBackwardFilter()238 backward_conv_dnums.add_output_spatial_dimensions(kernel_spatial_dims[i]); in MatchBackwardFilter()
362 std::vector<int64> kernel_spatial_dims(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local373 kernel_spatial_dims[i] = i; in MakeXlaBackpropInputConvOp()381 xla::XlaOp mirrored_weights = xla::Rev(filter, kernel_spatial_dims); in MakeXlaBackpropInputConvOp()
1701 std::vector<int64> kernel_spatial_dims(num_spatial_dims); in InferConvolveShape() local1703 kernel_spatial_dims[i] = rhs.dimensions(dnums.kernel_spatial_dimensions(i)); in InferConvolveShape()1765 if (kernel_spatial_dims != window_dims) { in InferConvolveShape()