HomeSort by relevance Sort by last modified time
    Searched refs:cell_state_out_ (Results 1 - 2 of 2) sorted by null

  /frameworks/ml/nn/common/operations/
LSTM.cpp 75 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor);
399 GetBuffer<float>(cell_state_out_));
408 GetBuffer<float>(cell_state_out_));
412 GetBuffer<float>(cell_state_out_));
415 tflite::tensor_utils::ClipVector(GetBuffer<float>(cell_state_out_), n_batch * n_cell,
416 params_.cell_clip_, GetBuffer<float>(cell_state_out_));
423 GetBuffer<float>(cell_state_out_), n_batch, output_gate_scratch);
427 tflite::tensor_utils::ApplyActivationToVector(GetBuffer<float>(cell_state_out_),
LSTM.h 140 RunTimeOperandInfo *cell_state_out_; member in class:android::nn::LSTMCell

Completed in 69 milliseconds