Home
last modified time | relevance | path

Searched refs:cellStateInCurrentTimeStep (Results 1 – 1 of 1) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSTM.cpp428 std::vector<float> cellStateInCurrentTimeStep(cell_state_in_buffer, in LSTMEvalFloat32() local
453 outputStateInCurrentTimeStep.data(), cellStateInCurrentTimeStep.data(), in LSTMEvalFloat32()
465 cellStateInCurrentTimeStep.assign(cell_state_out_buffer, in LSTMEvalFloat32()
665 std::vector<float> cellStateInCurrentTimeStep(batchSize * numCells); in LSTMEvalFloat16() local
666 convertFloat16ToFloat32(cell_state_in_buffer, &cellStateInCurrentTimeStep); in LSTMEvalFloat16()
695 cellStateInCurrentTimeStep.data(), input_layer_norm_weights_float32.data(), in LSTMEvalFloat16()
706 cellStateInCurrentTimeStep = cell_state_out_float32; in LSTMEvalFloat16()