Home
last modified time | relevance | path

Searched defs:SetWeights (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn_test.cc198 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
Dunidirectional_sequence_rnn_test.cc210 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
Dfully_connected_test.cc204 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
Dsvdf_test.cc212 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anon2a1e039b0111::HybridSVDFOpModel
Dlstm_test.cc285 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anonb3707f480111::HybridLSTMOpModel
1663 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anonb3707f480111::HybridLayerNormLSTMOpModel
Dunidirectional_sequence_lstm_test.cc255 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anon67ed54f90111::HybridUnidirectionalLSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc559 void SetWeights(std::initializer_list<float> data) { in SetWeights() function in tflite::__anon0c68fcb70111::FullyConnectedOpModel
1925 void SetWeights(std::initializer_list<float> f) { in SetWeights() function in tflite::__anon0c68fcb70111::RNNOpModel