Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Denvironment.h47 CLDevice* GetDevicePtr() { return &device_; } in GetDevicePtr() function
48 const CLDevice* GetDevicePtr() const { return &device_; } in GetDevicePtr() function
Denvironment.cc123 GetDevicePtr()->DisableOneLayerTextureArray(); in Init()
Dinference_context.cc142 creation_context.device = env->GetDevicePtr(); in InitFromGraph()
214 creation_context.device = env->GetDevicePtr(); in RestoreDeserialized()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dcl_test.cc43 return env_.GetDevicePtr()->GetInfo(); in GetGpuInfo()
52 creation_context.device = env_.GetDevicePtr(); in ExecuteGPUOperation()
Dcl_test.h68 creation_context_.device = env_.GetDevicePtr(); in SetUp()
Dconverter.cc564 impl->SetGpuInfo(environment_->GetDevicePtr()->GetInfo()); in MakeConverter()