Searched refs:kernel_spatial_dim (Results 1 – 5 of 5) sorted by relevance
77 int64_t kernel_spatial_dim = dim_nums.kernel_spatial_dimensions(i); in ExpandInstruction() local83 removed_kernel_dimensions.push_back(kernel_spatial_dim); in ExpandInstruction()88 new_dim_nums.add_kernel_spatial_dimensions(kernel_spatial_dim); in ExpandInstruction()
601 const int64_t kernel_spatial_dim = in HandleConvolution() local603 const int64_t kernel_limit = rhs_shape.dimensions(kernel_spatial_dim); in HandleConvolution()
1053 int64_t kernel_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local1065 custom_call_conv->mutable_operand(1), {}, kernel_spatial_dim); in RewriteDynamicConvolutionKernelGrad()1067 gradients = PadWithScalar(gradients, kernel_spatial_dim, in RewriteDynamicConvolutionKernelGrad()
325 const int64_t kernel_spatial_dim = dim_numbers.kernel_spatial_dimensions( in IsForwardWindowDilatedConv() local330 return convolution->operand(1)->shape().dimensions(kernel_spatial_dim) < in IsForwardWindowDilatedConv()3052 const int64_t kernel_spatial_dim = dim_numbers.kernel_spatial_dimensions( in GetConvolutionDetails() local3054 int64_t kernel_spatial_dim_size = kernel_shape.dimensions(kernel_spatial_dim); in GetConvolutionDetails()
216 auto kernel_spatial_dim = in HasCanonicalDimensionNumbers() local225 (*kernel_spatial_dim++).getZExtValue() != i) { in HasCanonicalDimensionNumbers()