| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_rnn_test.cc | 210 void SetWeights(std::initializer_list<float> f) { in SetWeights() function in tflite::__anona6a394470111::UnidirectionalRNNOpModel
258 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anona6a394470111::HybridUnidirectionalRNNOpModel
266 void SetWeights(std::initializer_list<float> f) { SetWeights(weights_, f); } in SetWeights() function in tflite::__anona6a394470111::HybridUnidirectionalRNNOpModel
269 SetWeights(recurrent_weights_, f); in SetRecurrentWeights()
279 rnn.SetWeights(rnn_weights); in TEST()
304 rnn.SetWeights(rnn_weights); in TEST()
330 rnn.SetWeights(rnn_weights); in TEST()
356 rnn.SetWeights(rnn_weights); in TEST()
385 rnn.SetWeights(rnn_weights); in TEST()
415 rnn.SetWeights(rnn_weights); in TEST()
|
| D | basic_rnn_test.cc | 198 void SetWeights(std::initializer_list<float> f) { in SetWeights() function in tflite::__anonaa6237b70111::RNNOpModel
243 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anonaa6237b70111::HybridRNNOpModel
251 void SetWeights(std::initializer_list<float> f) { SetWeights(weights_, f); } in SetWeights() function in tflite::__anonaa6237b70111::HybridRNNOpModel
254 SetWeights(recurrent_weights_, f); in SetRecurrentWeights()
260 rnn.SetWeights(rnn_weights); in TEST()
287 rnn.SetWeights(rnn_weights); in TEST()
315 rnn.SetWeights(rnn_weights); in TEST()
|
| D | fully_connected_test.cc | 204 void SetWeights(const std::vector<float>& f) { PopulateTensor(weights_, f); } in SetWeights() function in tflite::__anonc847e4c60111::FloatFullyConnectedOpModel
224 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::__anonc847e4c60111::QuantizedFullyConnectedOpModel
304 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::__anonc847e4c60111::HybridFullyConnectedOpModel
375 m.SetWeights({ in TEST_P()
395 m.SetWeights({ in TEST_P()
417 m.SetWeights({ in TEST()
438 m.SetWeights<uint8_t>({ in TEST_P()
468 m.SetWeights<int8_t>({ in TEST_P()
495 m.SetWeights<uint8_t>({ in TEST_P()
526 m.SetWeights<int8_t>({ in TEST_P()
[all …]
|
| D | lstm_test.cc | 285 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anonb3707f480111::HybridLSTMOpModel
294 SetWeights(input_to_input_weights_, f); in SetInputToInputWeights()
298 SetWeights(input_to_forget_weights_, f); in SetInputToForgetWeights()
302 SetWeights(input_to_cell_weights_, f); in SetInputToCellWeights()
306 SetWeights(input_to_output_weights_, f); in SetInputToOutputWeights()
310 SetWeights(recurrent_to_input_weights_, f); in SetRecurrentToInputWeights()
314 SetWeights(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
318 SetWeights(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
322 SetWeights(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
326 SetWeights(cell_to_input_weights_, f); in SetCellToInputWeights()
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 255 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anon67ed54f90111::HybridUnidirectionalLSTMOpModel
264 SetWeights(input_to_input_weights_, f); in SetInputToInputWeights()
268 SetWeights(input_to_forget_weights_, f); in SetInputToForgetWeights()
272 SetWeights(input_to_cell_weights_, f); in SetInputToCellWeights()
276 SetWeights(input_to_output_weights_, f); in SetInputToOutputWeights()
280 SetWeights(recurrent_to_input_weights_, f); in SetRecurrentToInputWeights()
284 SetWeights(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
288 SetWeights(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
292 SetWeights(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
296 SetWeights(cell_to_input_weights_, f); in SetCellToInputWeights()
[all …]
|
| D | svdf_test.cc | 212 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anon2a1e039b0111::HybridSVDFOpModel
221 SetWeights(weights_feature_, f); in SetWeightsFeature()
225 SetWeights(weights_time_, f); in SetWeightsTime()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 559 void SetWeights(std::initializer_list<float> data) { in SetWeights() function in tflite::__anon0c68fcb70111::FullyConnectedOpModel
585 m.SetWeights({ in TEST()
606 m.SetWeights({ in TEST()
1925 void SetWeights(std::initializer_list<float> f) { in SetWeights() function in tflite::__anon0c68fcb70111::RNNOpModel
1962 rnn.SetWeights(rnn_weights); in TEST()
|