Searched refs:kernel_spatial_dim (Results 1 – 4 of 4) sorted by relevance
77 int64 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()
600 const int64 kernel_spatial_dim = in HandleConvolution() local602 const int64 kernel_limit = rhs_shape.dimensions(kernel_spatial_dim); in HandleConvolution()
952 int64 kernel_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local964 custom_call_conv->mutable_operand(1), {}, kernel_spatial_dim); in RewriteDynamicConvolutionKernelGrad()966 gradients = PadWithScalar(gradients, kernel_spatial_dim, in RewriteDynamicConvolutionKernelGrad()
308 auto kernel_spatial_dim = in matchAndRewrite() local317 (*kernel_spatial_dim++).getZExtValue() != i) in matchAndRewrite()