Home
last modified time | relevance | path

Searched refs:KernelMetadata (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()
42 void KernelMetadata::set_registers_per_thread(int registers_per_thread) { in set_registers_per_thread()
47 bool KernelMetadata::shared_memory_bytes(int *shared_memory_bytes) const { in shared_memory_bytes()
56 void KernelMetadata::set_shared_memory_bytes(int shared_memory_bytes) { in set_shared_memory_bytes()
Dkernel.h100 class KernelMetadata {
102 KernelMetadata() in KernelMetadata() function
169 void set_metadata(const KernelMetadata &metadata) { metadata_ = metadata; } in set_metadata()
171 const KernelMetadata &metadata() const { return metadata_; } in metadata()
194 KernelMetadata metadata_;
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h265 KernelMetadata* kernel_metadata);
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc265 KernelMetadata kernel_metadata; in GetKernel()
275 KernelMetadata* kernel_metadata) { in GetKernelMetadata()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc318 KernelMetadata kernel_metadata; in GetKernel()
403 KernelMetadata* kernel_metadata) { in GetKernelMetadata()