Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drenamed_device.h96 PerOpGpuDevice* MakeGpuDevice() override { in MakeGpuDevice() function
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice_base.h216 virtual PerOpGpuDevice* MakeGpuDevice() { return nullptr; } in MakeGpuDevice() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc312 PerOpGpuDevice* MakeGpuDevice() override { in MakeGpuDevice() function in tensorflow::__anonbb96d3be0311::TfCallbackDevice
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc1171 PerOpGpuDevice* BaseGPUDevice::MakeGpuDevice() { in MakeGpuDevice() function in tensorflow::BaseGPUDevice