Home
last modified time | relevance | path

Searched refs:shared_memory_bytes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dkernel.cc46 bool KernelMetadata::shared_memory_bytes(int *shared_memory_bytes) const { in shared_memory_bytes() function in stream_executor::KernelMetadata
48 *shared_memory_bytes = shared_memory_bytes_; in shared_memory_bytes()
55 void KernelMetadata::set_shared_memory_bytes(int shared_memory_bytes) { in set_shared_memory_bytes() argument
56 shared_memory_bytes_ = shared_memory_bytes; in set_shared_memory_bytes()
Dkernel.h115 bool shared_memory_bytes(int *shared_memory_bytes) const;
119 void set_shared_memory_bytes(int shared_memory_bytes);
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc453 if (!kernel.metadata().shared_memory_bytes(&smem_per_block)) { in VlogOccupancyInfo()