Home
last modified time | relevance | path

Searched refs:UnloadKernel (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dkernel.cc80 parent_->UnloadKernel(this); in ~KernelBase()
Dstream_executor_internal.h197 virtual void UnloadKernel(const KernelBase *kernel) {} in UnloadKernel() function
Dstream_executor_pimpl.cc239 void StreamExecutor::UnloadKernel(const KernelBase *kernel) { in UnloadKernel() function in stream_executor::StreamExecutor
240 implementation_->UnloadKernel(kernel); in UnloadKernel()
Dstream_executor_pimpl.h108 void UnloadKernel(const KernelBase *kernel);
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h67 void UnloadKernel(const KernelBase* kernel) override;
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc139 void GpuExecutor::UnloadKernel(const KernelBase* kernel) { in UnloadKernel() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc344 void GpuExecutor::UnloadKernel(const KernelBase* kernel) { in UnloadKernel() function in stream_executor::gpu::GpuExecutor