Home
last modified time | relevance | path

Searched defs:GetContextHandle (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tensor.cc104 TFE_Context* GetContextHandle(PyObject* py_context) { in GetContextHandle() function
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_driver.cc366 /* static */ hipCtx_t GpuDriver::GetContextHandle(GpuContext* context) { in GetContextHandle() function in stream_executor::gpu::GpuDriver
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_driver.cc424 /* static */ CUcontext GpuDriver::GetContextHandle(GpuContext* context) { in GetContextHandle() function in stream_executor::gpu::GpuDriver