Searched defs:thread_dims (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_gpu_executor.cc | 298 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
353 const ThreadDim& thread_dims, in CalculateOccupancy()
363 const ThreadDim& thread_dims, in CompareOccupancy()
422 const ThreadDim& thread_dims, in VlogOccupancyInfo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 441 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
482 const ThreadDim& thread_dims, in VlogOccupancyInfo()
526 const ThreadDim& thread_dims, in CalculateOccupancy()
543 const ThreadDim& thread_dims, in CompareOccupancy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | trace_listener.h | 49 virtual void LaunchSubmit(Stream* stream, const ThreadDim& thread_dims, in LaunchSubmit()
|
| H A D | stream_executor_internal.h | 187 virtual port::Status Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
|
| H A D | stream.h | 1623 inline port::Status Stream::ThenLaunch(ThreadDim thread_dims, in ThenLaunch()
|
| H A D | stream_executor_pimpl.cc | 475 const ThreadDim& thread_dims, in Launch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| H A D | host_gpu_executor.h | 57 port::Status Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executor.h | 63 port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
|