Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn_test.cc197 void SetWeights(std::initializer_list<float> f) { in SetWeights() function in tflite::__anon2c82b85c0111::RNNOpModel
244 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anon2c82b85c0111::HybridRNNOpModel
252 void SetWeights(std::initializer_list<float> f) { SetWeights(weights_, f); } in SetWeights() function in tflite::__anon2c82b85c0111::HybridRNNOpModel
Dunidirectional_sequence_rnn_test.cc211 void SetWeights(std::initializer_list<float> f) { in SetWeights() function in tflite::__anona0a7fc2c0111::UnidirectionalRNNOpModel
261 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anona0a7fc2c0111::HybridUnidirectionalRNNOpModel
269 void SetWeights(std::initializer_list<float> f) { SetWeights(weights_, f); } in SetWeights() function in tflite::__anona0a7fc2c0111::HybridUnidirectionalRNNOpModel
Dfully_connected_test.cc266 void SetWeights(const std::vector<float>& f) { PopulateTensor(weights_, f); } in SetWeights() function in tflite::__anon1e118d2b0111::FloatFullyConnectedOpModel
305 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::__anon1e118d2b0111::QuantizedFullyConnectedOpModel
379 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::__anon1e118d2b0111::PerChannelQuantizedFullyConnectedOpModel
439 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::__anon1e118d2b0111::HybridFullyConnectedOpModel
Dbatch_matmul_test.cc415 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::__anon94d4a8fe0111::HybridBatchMatMulOpModel
946 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::__anon94d4a8fe0111::QuantizedBatchMatMulOpModel
Dsvdf_test.cc214 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anon10ad49200111::HybridSVDFOpModel
Dunidirectional_sequence_lstm_test.cc49 void SetWeights(int weights_idx, const std::vector<float>& f) { in SetWeights() function in tflite::__anona27eb97e0111::HybridUnidirectionalLSTMOpModel
Dlstm_test.cc284 void SetWeights(int index, const std::vector<float>& data) { in SetWeights() function in tflite::__anon99ffc4cd0111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dfully_connected_op_builder.cc36 void FullyConnectedOpBuilder::SetWeights(TfLiteTensor* weights) { in SetWeights() function in tflite::delegates::coreml::FullyConnectedOpBuilder
Dconvolution_op_builder.cc37 void ConvolutionOpBuilder::SetWeights(TfLiteTensor* weights) { in SetWeights() function in tflite::delegates::coreml::ConvolutionOpBuilder
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dmatmul_test.cc75 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::FullyConnectedOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_signed_quantization_test.cc106 void SetWeights(const std::vector<float>& data) { in SetWeights() function in tflite::__anon59489a3b0111::HybridFullyConnectedOpModel
Dnnapi_delegate_test.cc1299 void SetWeights(std::initializer_list<float> data) { in SetWeights() function in tflite::__anon4fbea17c0111::FullyConnectedOpModel
2836 void SetWeights(std::initializer_list<float> f) { in SetWeights() function in tflite::__anon4fbea17c0111::RNNOpModel