Searched refs:SetFromTfLite (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/delegates/flex/ |
D | buffer_map_test.cc | 82 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()
|
D | buffer_map.h | 57 void SetFromTfLite(int tensor_index, const TfLiteTensor* tensor);
|
D | buffer_map.cc | 147 void BufferMap::SetFromTfLite(int tensor_index, const TfLiteTensor* tensor) { in SetFromTfLite() function in tflite::flex::BufferMap
|
D | kernel.cc | 469 buffer_map->SetFromTfLite(tensor_index, tensor); in Prepare() 525 buffer_map->SetFromTfLite(tensor_index, tensor); in Eval()
|