Searched refs:kernel_spatial_dims (Results 1 – 5 of 5) sorted by relevance
186 auto kernel_spatial_dims = conv_dnums.kernel_spatial_dimensions(); in MatchBackwardFilter() local227 if ((kernel_spatial_dims.empty() || in MatchBackwardFilter()228 conv->operand(1)->shape().dimensions(kernel_spatial_dims[0]) <= in MatchBackwardFilter()315 for (int i = 0; i < kernel_spatial_dims.size(); ++i) { in MatchBackwardFilter()316 backward_conv_dnums.add_output_spatial_dimensions(kernel_spatial_dims[i]); in MatchBackwardFilter()
109 llvm::SmallVector<int64_t, 4> kernel_spatial_dims( in ConvertConvDimensionNumbers() local121 Convert(kernel_spatial_dims, builder), in ConvertConvDimensionNumbers()
338 std::vector<int64> kernel_spatial_dims(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local360 kernel_spatial_dims[i] = i; in MakeXlaBackpropInputConvOp()372 filter = xla::Rev(filter, kernel_spatial_dims); in MakeXlaBackpropInputConvOp()
1741 std::vector<int64> kernel_spatial_dims(num_spatial_dims); in InferConvolveShape() local1743 kernel_spatial_dims[i] = rhs.dimensions(dnums.kernel_spatial_dimensions(i)); in InferConvolveShape()1803 if (kernel_spatial_dims != window_dims) { in InferConvolveShape()
4362 SmallVector<int64_t, num_spatial_dims> kernel_spatial_dims; in matchAndRewrite() local4374 kernel_spatial_dims.push_back(dim); in matchAndRewrite()4435 GetI64ElementsAttr(kernel_spatial_dims, &rewriter); in matchAndRewrite()