Searched defs:p_shape (Results 1 – 4 of 4) sorted by relevance
197 ShapeHandle p_shape; in LuShapeFn() local
134 Tensor* p_shape = nullptr; in StoreAllDecodedSequences() local
70 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellFpropWithEigen() local153 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellBpropWithEigen() local
396 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell_size}); in LSTMBlockCellBpropWithCUDA() local