Searched refs:SetFromTensorFlow (Results 1 – 4 of 4) sorted by relevance
197 TEST(BufferMapTest, SetFromTensorFlow) { in TEST() argument202 buffer_map.SetFromTensorFlow(0, t1); in TEST()220 buffer_map.SetFromTensorFlow(0, t1); in TEST()221 buffer_map.SetFromTensorFlow(0, t2); in TEST()234 buffer_map.SetFromTensorFlow(0, t1); in TEST()249 buffer_map.SetFromTensorFlow(0, t1); in TEST()
58 void SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor);
61 void BufferMap::SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor) { in SetFromTensorFlow() function in tflite::flex::BufferMap
363 shared_info->buffer_map->SetFromTensorFlow(outputs_.TfLiteIndex(i), in MaybePersistTensorflowOutputs()