Searched refs:input_sizes (Results 1 – 1 of 1) sorted by relevance
418 Eigen::array<TensorIndex, 2> input_sizes; in convolution() local419 input_sizes[0] = m_; in convolution()420 input_sizes[1] = n_; in convolution()421 TensorMap<Tensor<T, 2>, Eigen::Aligned> A(a_, input_sizes); in convolution()