Home
last modified time | relevance | path

Searched refs:thread_counts_per_block (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dlaunch_dimensions.h46 const Dim3D& thread_counts_per_block) in LaunchDimensions() argument
48 thread_counts_per_block_(thread_counts_per_block) {} in LaunchDimensions()
52 Dim3D thread_counts_per_block() const { return thread_counts_per_block_; } in thread_counts_per_block() function
Dparallel_loop_emitter.cc90 llvm_ir::AddRangeMetadata(0, launch_dimensions_.thread_counts_per_block().x, in EmitIndexAndSetExitBasicBlock()
98 index_type, launch_dimensions_.thread_counts_per_block().x), in EmitIndexAndSetExitBasicBlock()
116 index_type, launch_dimensions_.thread_counts_per_block().x * in EmitIndexAndSetExitBasicBlock()
Dlaunch_dimensions.cc30 LaunchDimensions::Dim3D thread_counts = launch_dims.thread_counts_per_block(); in operator <<()
Dbuffer_comparator.cc621 LaunchDimensions::Dim3D thread_counts = dim.thread_counts_per_block(); in DeviceCompare()
Dstream_executor_util.cc218 LaunchDimensions::Dim3D thread_counts = dims.thread_counts_per_block(); in ExecuteKernelOnStream()
Dir_emitter_unnested.cc173 launch_dims.thread_counts_per_block().x); in UpdateLaunchDimensions()