Searched defs:kernel_spatial_dims (Results 1 – 6 of 6) sorted by relevance
93 llvm::SmallVector<int64_t, 4> kernel_spatial_dims( in ConvertConvDimensionNumbers() local
52 llvm::ArrayRef<int64_t> kernel_spatial_dims; member
107 auto kernel_spatial_dims = conv_dnums.kernel_spatial_dimensions(); in MatchBackwardFilter() local
338 std::vector<int64_t> kernel_spatial_dims(attrs.num_spatial_dims); in MakeXlaBackpropInputConvOp() local
1743 std::vector<int64_t> kernel_spatial_dims(num_spatial_dims); in InferConvolveShape() local
5033 SmallVector<int64_t, 4> kernel_spatial_dims; in matchAndRewrite() local5169 SmallVector<int64_t, num_spatial_dims> kernel_spatial_dims; in matchAndRewrite() local