Home
last modified time | relevance | path

Searched refs:launch_dimensions_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc39 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
48 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
57 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
80 llvm_ir::AddRangeMetadata(0, launch_dimensions_.block_counts().x, in EmitIndexAndSetExitBasicBlock()
90 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()
117 launch_dimensions_.block_counts().x), in EmitIndexAndSetExitBasicBlock()
163 int64 total_threads = launch_dimensions_.launch_bound(); in EmitLoop()
Dkernel_thunk.cc67 launch_dimensions_ = launch_dims; in SetLaunchDimensions()
98 launch_dimensions = launch_dimensions_; in ExecuteOnStream()
Dkernel_thunk.h73 LaunchDimensions launch_dimensions_; variable
Dparallel_loop_emitter.h68 const LaunchDimensions launch_dimensions_;