Searched refs:TensorDim (Results 1 – 1 of 1) sorted by relevance
326 typedef Eigen::array<IndexType, 3> TensorDim; in contraction_batch() typedef328 TensorDim left_dims = {{m_batch, k_size, m_size}}; in contraction_batch()329 TensorDim right_dims = {{m_batch, n_size, k_size}}; in contraction_batch()330 TensorDim res_dims = {{m_batch, m_size, n_size}}; in contraction_batch()