Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dkernel.cc32 bool KernelMetadata::registers_per_thread(int *registers_per_thread) const { in registers_per_thread()
41 void KernelMetadata::set_registers_per_thread(int registers_per_thread) { in set_registers_per_thread()
46 bool KernelMetadata::shared_memory_bytes(int *shared_memory_bytes) const { in shared_memory_bytes()
55 void KernelMetadata::set_shared_memory_bytes(int shared_memory_bytes) { in set_shared_memory_bytes()
Dkernel.h101 class KernelMetadata {
103 KernelMetadata() in KernelMetadata() function
170 void set_metadata(const KernelMetadata &metadata) { metadata_ = metadata; } in set_metadata()
172 const KernelMetadata &metadata() const { return metadata_; } in metadata()
195 KernelMetadata metadata_;
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h263 KernelMetadata* kernel_metadata);
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc278 KernelMetadata kernel_metadata; in GetKernel()
286 KernelMetadata* kernel_metadata) { in GetKernelMetadata()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc308 KernelMetadata kernel_metadata; in GetKernel()
387 KernelMetadata* kernel_metadata) { in GetKernelMetadata()