Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dkernel.cc33 bool KernelMetadata::registers_per_thread(int *registers_per_thread) const { in registers_per_thread() function in stream_executor::KernelMetadata
35 *registers_per_thread = registers_per_thread_; in registers_per_thread()
42 void KernelMetadata::set_registers_per_thread(int registers_per_thread) { in set_registers_per_thread() argument
43 registers_per_thread_ = registers_per_thread; in set_registers_per_thread()
Dkernel.h106 bool registers_per_thread(int *registers_per_thread) const;
109 void set_registers_per_thread(int registers_per_thread);
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h78 uint64 registers_per_thread,
85 uint64 registers_per_thread,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc475 if (!kernel.metadata().registers_per_thread(&regs_per_thread)) { in VlogOccupancyInfo()
509 uint64 registers_per_thread, in CalculateOccupancy() argument
526 uint64 registers_per_thread, in CompareOccupancy() argument
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc345 uint64 registers_per_thread, in CalculateOccupancy() argument
355 uint64 registers_per_thread, in CompareOccupancy() argument