Searched refs:kernel_dim (Results 1 – 3 of 3) sorted by relevance
34 const Index kernel_dim = kernel_dims[i]; in IndexMapper() local35 const Index result_dim = input_dim - kernel_dim + 1; in IndexMapper()338 const Index kernel_dim = kernel_dims[i];339 const Index result_dim = input_dim - kernel_dim + 1;357 const Index kernel_dim = kernel_dims[i];358 const Index result_dim = input_dim - kernel_dim + 1;784 const Index kernel_dim = kernel_dims[i];785 const Index result_dim = input_dim - kernel_dim + 1;
134 int64 kernel_dim = in IsConvolutionKernelSmall() local138 if (kernel_dim >= output_dim) { in IsConvolutionKernelSmall()
1168 int kernel_dim = descriptor.dnums.kernel_spatial_dimensions(index); in EmitConvolutionThunkFromMlir() local1169 dim->set_size(descriptor.operand0_shape.dimensions(kernel_dim)); in EmitConvolutionThunkFromMlir()