Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.cpp471 std::vector<float> cellStateInCurrentTimeStep(cell_state_in_buffer, in LSTMEvalFloat32() local
496 outputStateInCurrentTimeStep.data(), cellStateInCurrentTimeStep.data(), in LSTMEvalFloat32()
508 cellStateInCurrentTimeStep.assign(cell_state_out_buffer, in LSTMEvalFloat32()
708 std::vector<float> cellStateInCurrentTimeStep(batchSize * numCells); in LSTMEvalFloat16() local
709 convertFloat16ToFloat32(cell_state_in_buffer, &cellStateInCurrentTimeStep); in LSTMEvalFloat16()
738 cellStateInCurrentTimeStep.data(), input_layer_norm_weights_float32.data(), in LSTMEvalFloat16()
749 cellStateInCurrentTimeStep = cell_state_out_float32; in LSTMEvalFloat16()