Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc199 ScopedActivateContext::ScopedActivateContext(GpuContext* cuda_context) { in ScopedActivateContext() function in stream_executor::gpu::ScopedActivateContext
238 ScopedActivateContext::~ScopedActivateContext() { in ~ScopedActivateContext()
504 ScopedActivateContext activation(context); in ContextGetSharedMemConfig()
512 ScopedActivateContext activation(context); in ContextSetSharedMemConfig()
524 ScopedActivateContext activation(context); in LaunchKernel()
540 ScopedActivateContext activation(context); in LoadCubin()
553 ScopedActivateContext activation(context); in LoadPtx()
626 ScopedActivateContext activation(context); in SynchronousMemsetUint8()
635 ScopedActivateContext activation(context); in SynchronousMemsetUint32()
644 ScopedActivateContext activation(context); in AsynchronousMemsetUint8()
[all …]
Dcuda_driver.h53 using ScopedActivateContext = gpu::ScopedActivateContext; variable
Dcuda_driver_test.cc57 ScopedActivateContext scope(&se_context1); in TEST()
66 ScopedActivateContext scope(&se_context1); in TEST()
Dcuda_asm_compiler.cc37 gpu::ScopedActivateContext activation(context); in LinkGpuAsm()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc175 ScopedActivateContext::ScopedActivateContext(GpuContext* context) { in ScopedActivateContext() function in stream_executor::gpu::ScopedActivateContext
207 ScopedActivateContext::~ScopedActivateContext() { in ~ScopedActivateContext()
382 ScopedActivateContext activation{context}; in ContextGetSharedMemConfig()
391 ScopedActivateContext activation{context}; in ContextSetSharedMemConfig()
404 ScopedActivateContext activation{context}; in LaunchKernel()
439 ScopedActivateContext activation{context}; in LoadHsaco()
460 ScopedActivateContext activation{context}; in SynchronousMemsetUint8()
469 ScopedActivateContext activation{context}; in SynchronousMemsetUint32()
480 ScopedActivateContext activation{context}; in AsynchronousMemsetUint8()
490 ScopedActivateContext activation{context}; in AsynchronousMemsetUint32()
[all …]
/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.h578 class ScopedActivateContext {
584 explicit ScopedActivateContext(GpuContext* context);
588 ~ScopedActivateContext();