Searched refs:kernel_sizes (Results 1 – 1 of 1) sorted by relevance
422 Eigen::array<TensorIndex, 2> kernel_sizes; in convolution() local423 kernel_sizes[0] = kernel_x; in convolution()424 kernel_sizes[1] = kernel_y; in convolution()425 TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, kernel_sizes); in convolution()