/external/tensorflow/tensorflow/lite/kernels/ |
optional_tensor_test.cc | 50 input_to_cell_weights_ = AddInput(TensorType_FLOAT32); 122 PopulateTensor(input_to_cell_weights_, f); 200 int input_to_cell_weights_; member in class:tflite::__anon46072::LSTMOpModel
|
unidirectional_sequence_lstm_test.cc | 55 input_to_cell_weights_ = AddInput(weights_type); 131 PopulateTensor(input_to_cell_weights_, f); 207 int input_to_cell_weights_; member in class:tflite::__anon46117::UnidirectionalLSTMOpModel 272 SetWeights(input_to_cell_weights_, f); 319 std::vector<float> input_to_cell_weights_; member in class:tflite::__anon46117::BaseLstmTest 406 input_to_cell_weights_ = {-0.50013041, 0.1370284, 0.11810488, 0.2013163, 494 lstm.SetInputToCellWeights(input_to_cell_weights_); 556 lstm.SetInputToCellWeights(input_to_cell_weights_); 623 lstm.SetInputToCellWeights(input_to_cell_weights_); 686 lstm.SetInputToCellWeights(input_to_cell_weights_); [all...] |
lstm_test.cc | 56 input_to_cell_weights_ = AddInput(weight_type); 141 PopulateTensor(input_to_cell_weights_, f); 232 int input_to_cell_weights_; member in class:tflite::__anon46063::LSTMOpModel 302 SetWeights(input_to_cell_weights_, f); 346 std::vector<float> input_to_cell_weights_; member in class:tflite::__anon46063::BaseLstmTest 405 input_to_cell_weights_ = {-0.50013041, 0.1370284, 0.11810488, 0.2013163, 488 lstm.SetInputToCellWeights(input_to_cell_weights_); 546 lstm.SetInputToCellWeights(input_to_cell_weights_); 605 lstm.SetInputToCellWeights(input_to_cell_weights_); 625 input_to_cell_weights_ = {-0.49770179, -0.27711356, -0.09624726 1740 std::vector<float> input_to_cell_weights_; member in class:tflite::__anon46063::BaseLayerNormLstmTest [all...] |
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
nnapi_delegate_test.cc | 2425 int input_to_cell_weights_; member in class:tflite::__anon45929::LSTMOpModel 2464 std::vector<float> input_to_cell_weights_; member in class:tflite::__anon45929::BaseLstmTest [all...] |