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.cc82 buffer_map.SetFromTfLite(0, t1.get()); in MakeTensor()
105 TEST(BufferMapTest, SetFromTfLite) { in TEST() argument
110 buffer_map.SetFromTfLite(0, t.get()); in TEST()
128 buffer_map.SetFromTfLite(0, t.get()); in TEST()
148 buffer_map.SetFromTfLite(0, t1.get()); in TEST()
149 buffer_map.SetFromTfLite(0, t2.get()); in TEST()
162 buffer_map.SetFromTfLite(0, t1.get()); in TEST()
163 buffer_map.SetFromTfLite(0, t2.get()); in TEST()
206 buffer_map.SetFromTfLite(0, t2.get()); in TEST()
219 buffer_map.SetFromTfLite(0, t2.get()); in TEST()
Dbuffer_map.h57 void SetFromTfLite(int tensor_index, const TfLiteTensor* tensor);
Dbuffer_map.cc147 void BufferMap::SetFromTfLite(int tensor_index, const TfLiteTensor* tensor) { in SetFromTfLite() function in tflite::flex::BufferMap
Dkernel.cc469 buffer_map->SetFromTfLite(tensor_index, tensor); in Prepare()
525 buffer_map->SetFromTfLite(tensor_index, tensor); in Eval()