Home
last modified time | relevance | path

Searched refs:shared_memory_per_block (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_device_info.h34 int shared_memory_per_block; member
Dbuffer_comparator.cc613 gpu_device_info.shared_memory_per_block = in DeviceCompare()
614 executor->GetDeviceDescription().shared_memory_per_block(); in DeviceCompare()
Dgpu_compiler.cc969 gpu_device_info.shared_memory_per_block = in GetGpuDeviceInfo()
970 stream_exec->GetDeviceDescription().shared_memory_per_block(); in GetGpuDeviceInfo()
Dir_emitter_unnested.cc2966 ir_emitter_context_->gpu_device_info().shared_memory_per_block; in EmitSortFromMlir()
2975 ir_emitter_context_->gpu_device_info().shared_memory_per_block); in EmitSortFromMlir()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dhlo_to_llvm_ir.cc49 gpu_device_info.shared_memory_per_block = 49152; in CompileAndPrintLlvmIr()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h79 uint64 shared_memory_per_block,
86 uint64 shared_memory_per_block,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc483 uint64 shared_memory_per_block, in CalculateOccupancy() argument
490 shared_memory_per_block, 0); in CalculateOccupancy()
500 uint64 shared_memory_per_block, in CompareOccupancy() argument
507 shared_memory_per_block, 0); in CompareOccupancy()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h144 int64_t 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/rocm/
Drocm_gpu_executor.cc350 uint64 shared_memory_per_block, in CalculateOccupancy() argument
360 uint64 shared_memory_per_block, in CompareOccupancy() argument