Home
last modified time | relevance | path

Searched refs:ScopedActivateContext (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc174 ScopedActivateContext::ScopedActivateContext(GpuContext* context) { in ScopedActivateContext() function in stream_executor::gpu::ScopedActivateContext
206 ScopedActivateContext::~ScopedActivateContext() { in ~ScopedActivateContext()
404 ScopedActivateContext activation{context}; in ContextGetSharedMemConfig()
420 ScopedActivateContext activation{context}; in ContextSetSharedMemConfig()
441 ScopedActivateContext activation{context}; in LaunchKernel()
479 ScopedActivateContext activation{context}; in LoadHsaco()
502 ScopedActivateContext activation{context}; in SynchronousMemsetUint8()
515 ScopedActivateContext activation{context}; in SynchronousMemsetUint32()
530 ScopedActivateContext activation{context}; in AsynchronousMemsetUint8()
546 ScopedActivateContext activation{context}; in AsynchronousMemsetUint32()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc206 ScopedActivateContext::ScopedActivateContext(GpuContext* cuda_context) { in ScopedActivateContext() function in stream_executor::gpu::ScopedActivateContext
230 ScopedActivateContext::~ScopedActivateContext() { in ~ScopedActivateContext()
516 ScopedActivateContext activation(context); in ContextGetSharedMemConfig()
533 ScopedActivateContext activation(context); in ContextSetSharedMemConfig()
555 ScopedActivateContext activation(context); in LaunchKernel()
575 ScopedActivateContext activation(context); in LoadCubin()
592 ScopedActivateContext activation(context); in LoadPtx()
665 ScopedActivateContext activation(context); in SynchronousMemsetUint8()
678 ScopedActivateContext activation(context); in SynchronousMemsetUint32()
692 ScopedActivateContext activation(context); in AsynchronousMemsetUint8()
[all …]
Dcuda_driver.h53 using ScopedActivateContext = gpu::ScopedActivateContext; variable
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_activation.h35 class ScopedActivateContext; variable
53 ScopedActivateContext* driver_scoped_activate_context_;
Dgpu_activation.cc31 new ScopedActivateContext{ExtractGpuContext(gpu_exec)}) {} in ScopedActivateExecutorContext()
38 delete static_cast<ScopedActivateContext*>(driver_scoped_activate_context_); in ~ScopedActivateExecutorContext()
Dgpu_driver.h506 class ScopedActivateContext {
512 explicit ScopedActivateContext(GpuContext* context);
516 ~ScopedActivateContext();