Home
last modified time | relevance | path

Searched refs:GpuFunctionHandle (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_kernel.h53 GpuFunctionHandle AsGpuFunctionHandle() const { in AsGpuFunctionHandle()
55 return const_cast<GpuFunctionHandle>(gpu_function_); in AsGpuFunctionHandle()
60 GpuFunctionHandle* gpu_function_ptr() { return &gpu_function_; } in gpu_function_ptr()
83 GpuFunctionHandle gpu_function_; // Wrapped CUDA kernel handle.
Dgpu_driver.h179 GpuFunctionHandle function,
184 static bool FuncSetCacheConfig(GpuFunctionHandle function,
203 static bool LaunchKernel(GpuContext* context, GpuFunctionHandle function,
234 GpuFunctionHandle* function);
497 GpuContext* context, GpuFunctionHandle kernel, int threads_per_block,
Dgpu_types.h49 using GpuFunctionHandle = hipFunction_t; variable
67 using GpuFunctionHandle = CUfunction;
Dgpu_executor.h80 const ThreadDim& thread_dims, GpuFunctionHandle func);
87 const ThreadDim& thread_dims, GpuFunctionHandle func);
315 std::set<GpuFunctionHandle> launched_kernels_
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc348 GpuFunctionHandle func) { in CalculateOccupancy()
358 GpuFunctionHandle func) { in CompareOccupancy()