Home
last modified time | relevance | path

Searched refs:FindBuffer (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/
Dgl_delegate.cc101 auto buffer = phwc4_objects_.FindBuffer(handle); in CopyFromBufferHandle()
114 auto buffer = phwc4_objects_.FindBuffer(handle); in CopyToBufferHandle()
194 auto external_buffer = bhwc_objects_.FindBuffer(tensor_index); in Prepare()
232 auto external_buffer = bhwc_objects_.FindBuffer(tensor_index); in Prepare()
290 auto external_object = bhwc_objects_.FindBuffer(ref.tensor_index); in Invoke()
297 phwc4_objects_.FindBuffer(id))); in Invoke()
314 auto external_object = bhwc_objects_.FindBuffer(ref.tensor_index); in Invoke()
320 phwc4_to_bhwc_.Convert(ref.shape, *phwc4_objects_.FindBuffer(id), in Invoke()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi.cc111 auto buffer = objects_->FindBuffer(id); in Execute()
134 auto buffer = objects_->FindBuffer(id); in Execute()
136 auto ref = refs_->FindBuffer(id); in Execute()
255 auto buffer = objects->FindBuffer(s.first); in NewRun()
Dobject_manager.h49 GlBuffer* FindBuffer(uint32_t id) const;
Dobject_manager.cc64 GlBuffer* ObjectManager::FindBuffer(uint32_t id) const { in FindBuffer() function in tflite::gpu::gl::ObjectManager
Druntime.cc152 auto ptr = objects->FindBuffer(id); in MakeBindingFunc()
189 auto ptr = objects->FindBuffer(id); in MakeLateBindingFunc()
195 auto ptr = objects->FindBuffer(id); in MakeLateBindingFunc()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dtest_util.cc125 CopyFromPHWC4Buffer(*objects.FindBuffer(output->id), &tensor)); in Invoke()