Home
last modified time | relevance | path

Searched defs:shared_memory_per_block (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_device_info.h34 int shared_memory_per_block; member
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.h155 int64 shared_memory_per_block() const { return shared_memory_per_block_; } in shared_memory_per_block() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc483 uint64 shared_memory_per_block, in CalculateOccupancy()
500 uint64 shared_memory_per_block, in CompareOccupancy()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc350 uint64 shared_memory_per_block, in CalculateOccupancy()
360 uint64 shared_memory_per_block, in CompareOccupancy()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h144 int64_t shared_memory_per_block; member