Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_device_info.h30 int shared_memory_per_block; member
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc352 uint64_t shared_memory_per_block, in CalculateOccupancy()
362 uint64_t shared_memory_per_block, in CompareOccupancy()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc525 uint64_t shared_memory_per_block, in CalculateOccupancy()
542 uint64_t shared_memory_per_block, in CompareOccupancy()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddevice_description.h285 int64_t shared_memory_per_block() const { return shared_memory_per_block_; } in shared_memory_per_block() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_decl.h144 int64_t shared_memory_per_block; member