Searched refs:shared_memory_bytes (Results 1 – 4 of 4) sorted by relevance
46 bool KernelMetadata::shared_memory_bytes(int *shared_memory_bytes) const { in shared_memory_bytes() function in stream_executor::KernelMetadata48 *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() argument56 shared_memory_bytes_ = shared_memory_bytes; in set_shared_memory_bytes()
115 bool shared_memory_bytes(int *shared_memory_bytes) const;119 void set_shared_memory_bytes(int shared_memory_bytes);
384 unsigned shared_memory_bytes = block.x * block.y * sizeof(Treducevec);388 grid, block, shared_memory_bytes, d.stream(), nouter, ninner_vec,
495 if (!kernel.metadata().shared_memory_bytes(&smem_per_block)) { in VlogOccupancyInfo()