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()
618 const int64_t kernel_spatial_dim = in GetConvolutionFlops() local620 const int64_t kernel_limit = rhs_shape.dimensions(kernel_spatial_dim); in GetConvolutionFlops()
1109 int64_t kernel_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local1121 custom_call_conv->mutable_operand(1), {}, kernel_spatial_dim); in RewriteDynamicConvolutionKernelGrad()1123 gradients = PadWithScalar(gradients, kernel_spatial_dim, in RewriteDynamicConvolutionKernelGrad()
385 const int64_t kernel_spatial_dim = dim_numbers.kernel_spatial_dimensions( in IsForwardWindowDilatedConv() local390 return convolution->operand(1)->shape().dimensions(kernel_spatial_dim) < in IsForwardWindowDilatedConv()3548 const int64_t kernel_spatial_dim = dim_numbers.kernel_spatial_dimensions( in GetConvolutionDetails() local3550 int64_t kernel_spatial_dim_size = kernel_shape.dimensions(kernel_spatial_dim); in GetConvolutionDetails()
477 auto kernel_spatial_dim = dnums.getKernelSpatialDimensions(); in IsSamePadding() local487 kernel_spatial_dim[i]), in IsSamePadding()