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