Home
last modified time | relevance | path

Searched refs:SetFromTfLite (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_test.cc76 buffer_map.SetFromTfLite(0, t1.get()); in MakeTensor()
99 TEST(BufferMapTest, SetFromTfLite) { in TEST() argument
104 buffer_map.SetFromTfLite(0, t.get()); in TEST()
122 buffer_map.SetFromTfLite(0, t.get()); in TEST()
142 buffer_map.SetFromTfLite(0, t1.get()); in TEST()
143 buffer_map.SetFromTfLite(0, t2.get()); in TEST()
156 buffer_map.SetFromTfLite(0, t1.get()); in TEST()
157 buffer_map.SetFromTfLite(0, t2.get()); in TEST()
200 buffer_map.SetFromTfLite(0, t2.get()); in TEST()
213 buffer_map.SetFromTfLite(0, t2.get()); in TEST()
Dbuffer_map.h62 void SetFromTfLite(int tensor_index, const TfLiteTensor* tensor);
Dbuffer_map.cc157 void BufferMap::SetFromTfLite(int tensor_index, const TfLiteTensor* tensor) { in SetFromTfLite() function in tflite::flex::BufferMap
Dkernel.cc460 buffer_map->SetFromTfLite(tensor_index, tensor); in Prepare()
606 buffer_map->SetFromTfLite(tensor_index, tensor); in Eval()