Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dvirtual_device.cc92 Status VirtualDevice::TryGetDeviceContext(DeviceContext** out_context) { in TryGetDeviceContext() function in tensorflow::VirtualDevice
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice.h149 virtual Status TryGetDeviceContext(DeviceContext** out_context) { in TryGetDeviceContext() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drenamed_device.h138 Status TryGetDeviceContext(DeviceContext** out_context) override { in TryGetDeviceContext() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.cc415 Status XlaDevice::TryGetDeviceContext(DeviceContext** out_context) { in TryGetDeviceContext() function in tensorflow::XlaDevice