Home
last modified time | relevance | path

Searched defs:thread_dims (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dtrace_listener.h49 virtual void LaunchSubmit(Stream* stream, const ThreadDim& thread_dims, in LaunchSubmit()
Dstream_executor_internal.h190 virtual port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
Dstream.h2119 inline Stream &Stream::ThenLaunch(ThreadDim thread_dims, BlockDim block_dims, in ThenLaunch()
Dstream_executor_pimpl.cc465 const ThreadDim &thread_dims, in Launch()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h57 port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc400 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
440 const ThreadDim& thread_dims, in VlogOccupancyInfo()
484 const ThreadDim& thread_dims, in CalculateOccupancy()
501 const ThreadDim& thread_dims, in CompareOccupancy()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc296 port::Status GpuExecutor::Launch(Stream* stream, const ThreadDim& thread_dims, in Launch()
351 const ThreadDim& thread_dims, in CalculateOccupancy()
361 const ThreadDim& thread_dims, in CompareOccupancy()
420 const ThreadDim& thread_dims, in VlogOccupancyInfo()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h64 port::Status Launch(Stream *stream, const ThreadDim &thread_dims, in Launch()
/external/llvm-project/polly/lib/External/ppcg/
Dcuda.c394 const char *thread_dims[] = { "threadIdx.x", "threadIdx.y", in print_kernel_iterators() local