HomeSort by relevance Sort by last modified time
    Searched refs:CUDAKernel (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_kernel.h 48 class CUDAKernel : public internal::KernelInterface {
50 CUDAKernel() : cuda_function_(nullptr), arity_(0),
55 ~CUDAKernel() override {}
73 // (more-or-less equivalent to a CUDAKernel). We support this via the below
116 inline const CUDAKernel *AsCUDAKernel(const KernelBase *kernel) {
117 return static_cast<const CUDAKernel *>(kernel->implementation());
122 inline CUDAKernel *AsCUDAKernel(KernelBase *kernel) {
123 return static_cast<CUDAKernel *>(kernel->implementation());
cuda_gpu_executor.cc 212 CUDAKernel *cuda_kernel = AsCUDAKernel(kernel);
330 bool CUDAExecutor::GetKernelMetadata(CUDAKernel *cuda_kernel,
355 const CUDAKernel *cuda_kernel = AsCUDAKernel(&kernel);
835 return std::unique_ptr<internal::KernelInterface>(new CUDAKernel());
    [all...]
cuda_gpu_executor.h 227 bool GetKernelMetadata(CUDAKernel *cuda_kernel,

Completed in 154 milliseconds