| /external/tensorflow/tensorflow/lite/kernels/ |
| lstm_test.cc | 75 cell_to_forget_weights_ = AddInput(weight_type);
79 cell_to_forget_weights_ = AddNullInput();
169 PopulateTensor(cell_to_forget_weights_, f);
241 int cell_to_forget_weights_; member in class:tflite::__anon46063::LSTMOpModel
330 SetWeights(cell_to_forget_weights_, f);
358 std::vector<float> cell_to_forget_weights_; member in class:tflite::__anon46063::BaseLstmTest
658 cell_to_forget_weights_ = {0.47485286, -0.51955009, -0.24458408,
720 lstm.SetCellToForgetWeights(cell_to_forget_weights_);
779 lstm.SetCellToForgetWeights(cell_to_forget_weights_);
838 lstm.SetCellToForgetWeights(cell_to_forget_weights_);
1752 std::vector<float> cell_to_forget_weights_; member in class:tflite::__anon46063::BaseLayerNormLstmTest
[all...] |
| optional_tensor_test.cc | 69 cell_to_forget_weights_ = AddInput(TensorType_FLOAT32);
73 cell_to_forget_weights_ = AddNullInput();
150 PopulateTensor(cell_to_forget_weights_, f);
209 int cell_to_forget_weights_; member in class:tflite::__anon46072::LSTMOpModel
|
| unidirectional_sequence_lstm_test.cc | 74 cell_to_forget_weights_ = AddInput(weights_type);
78 cell_to_forget_weights_ = AddNullInput();
159 PopulateTensor(cell_to_forget_weights_, f);
216 int cell_to_forget_weights_; member in class:tflite::__anon46117::UnidirectionalLSTMOpModel
300 SetWeights(cell_to_forget_weights_, f);
331 std::vector<float> cell_to_forget_weights_; member in class:tflite::__anon46117::BaseLstmTest
739 cell_to_forget_weights_ = {0.47485286, -0.51955009, -0.24458408,
806 lstm.SetCellToForgetWeights(cell_to_forget_weights_);
869 lstm.SetCellToForgetWeights(cell_to_forget_weights_);
[all...] |
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| nnapi_delegate_test.cc | 2434 int cell_to_forget_weights_; member in class:tflite::__anon45929::LSTMOpModel
2476 std::vector<float> cell_to_forget_weights_; member in class:tflite::__anon45929::BaseLstmTest
[all...] |
