Home
last modified time | relevance | path

Searched defs:ReinitializeGpuDevice (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drenamed_device.h100 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device, in ReinitializeGpuDevice() function
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice_base.h223 virtual Status ReinitializeGpuDevice(OpKernelContext* /*context*/, in ReinitializeGpuDevice() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc320 Status ReinitializeGpuDevice(OpKernelContext* context, PerOpGpuDevice* device, in ReinitializeGpuDevice() function in tensorflow::__anonbb96d3be0311::TfCallbackDevice
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc1175 Status BaseGPUDevice::ReinitializeGpuDevice(OpKernelContext* context, in ReinitializeGpuDevice() function in tensorflow::BaseGPUDevice