Searched defs:p_broadcast_shape (Results 1 – 2 of 2) sorted by relevance
71 Eigen::array<Eigen::DenseIndex, 2> p_broadcast_shape({cell.batch_size(), 1}); in LSTMBlockCellFpropWithEigen() local150 Eigen::array<Eigen::DenseIndex, 2> p_broadcast_shape({cell.batch_size(), 1}); in LSTMBlockCellBpropWithEigen() local
390 Eigen::array<Eigen::DenseIndex, 2> p_broadcast_shape({batch_size, 1}); in LSTMBlockCellBpropWithCUDA() local