Searched refs:GpuFunctionHandle (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_kernel.h | 53 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.
|
| D | gpu_driver.h | 236 GpuFunctionHandle function,
241 static port::Status FuncSetCacheConfig(GpuFunctionHandle function,
261 GpuContext* context, GpuFunctionHandle function, unsigned int grid_dim_x,
290 GpuFunctionHandle* function);
569 GpuContext* context, GpuFunctionHandle kernel, int threads_per_block,
|
| D | gpu_types.h | 49 using GpuFunctionHandle = hipFunction_t; variable
67 using GpuFunctionHandle = CUfunction;
|
| D | gpu_executor.h | 80 const ThreadDim& thread_dims, GpuFunctionHandle func);
87 const ThreadDim& thread_dims, GpuFunctionHandle func);
314 std::set<GpuFunctionHandle> launched_kernels_
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 352 GpuFunctionHandle func) { in CalculateOccupancy()
362 GpuFunctionHandle func) { in CompareOccupancy()
|