Home
last modified time | relevance | path

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

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