Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/contrib/lite/kernels/ | |
optional_tensor_test.cc | 82 forget_gate_bias_ = AddInput(TensorType_FLOAT32); 161 PopulateTensor(forget_gate_bias_, f); 228 int forget_gate_bias_; member in class:tflite::__anon39267::LSTMOpModel |
lstm_test.cc | 84 forget_gate_bias_ = AddInput(TensorType_FLOAT32); 163 PopulateTensor(forget_gate_bias_, f); 226 int forget_gate_bias_; member in class:tflite::__anon39264::LSTMOpModel [all...] |
unidirectional_sequence_lstm_test.cc | 87 forget_gate_bias_ = AddInput(TensorType_FLOAT32); 167 PopulateTensor(forget_gate_bias_, f); 231 int forget_gate_bias_; member in class:tflite::__anon39285::UnidirectionalLSTMOpModel [all...] |
/frameworks/ml/nn/common/operations/ | |
LSTM.h | 129 const RunTimeOperandInfo *forget_gate_bias_; member in class:android::nn::LSTMCell |
LSTM.cpp | 59 forget_gate_bias_ = GetInput(operation, operands, kForgetGateBiasTensor); 335 tflite::tensor_utils::VectorBatchVectorAssign(GetBuffer<float>(forget_gate_bias_), |