Home
last modified time | relevance | path

Searched refs:GetContextHandle (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc104 TFE_Context* GetContextHandle(PyObject* py_context) { in GetContextHandle() function
524 auto* handle = tensorflow::ConvertToEagerTensor(GetContextHandle(py_context), in EagerTensor_init()
692 self->handle, GetContextHandle(self->context), device_name, in EagerTensor_copy_to_device()
1150 TFE_Context* ctx = GetContextHandle(py_context); in TFE_Py_TensorShapeSlice()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dgpu_driver.h232 static GpuContextHandle GetContextHandle(GpuContext* context);
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc366 /* static */ hipCtx_t GpuDriver::GetContextHandle(GpuContext* context) { in GetContextHandle() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_driver.cc424 /* static */ CUcontext GpuDriver::GetContextHandle(GpuContext* context) { in GetContextHandle() function in stream_executor::gpu::GpuDriver