Searched refs:SetModelWeight (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/ |
| H A D | predict_test.cc | 68 void SetModelWeight(std::initializer_list<float> data) { function in class:tflite::ops::custom::__anon25970::PredictOpModel
113 m.SetModelWeight({0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2});
124 m.SetModelWeight({0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2});
135 m.SetModelWeight({0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2});
146 m.SetModelWeight({0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2, 0.1, 0.2});
157 m.SetModelWeight({0.1, 0, 0.1, 0, 0.1, 0, 0.1, 0, 0.1, 0});
168 m.SetModelWeight({0.1, 0, 0.1, 0, 0.1, 0, 0.1, 0, 0, 0});
|
Completed in 399 milliseconds