Searched refs:SetFromTensorFlow (Results 1 – 4 of 4) sorted by relevance
163 TEST(BufferMapTest, SetFromTensorFlow) { in TEST() argument168 buffer_map.SetFromTensorFlow(0, t1); in TEST()185 buffer_map.SetFromTensorFlow(0, t1); in TEST()186 buffer_map.SetFromTensorFlow(0, t2); in TEST()199 buffer_map.SetFromTensorFlow(0, t1); in TEST()214 buffer_map.SetFromTensorFlow(0, t1); in TEST()
58 void SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor);
181 void BufferMap::SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor) { in SetFromTensorFlow() function in tflite::flex::BufferMap
309 buffer_map->SetFromTensorFlow(outputs_.TfLiteIndex(i), *tensor); in PersistEagerOutputs()