Home
last modified time | relevance | path

Searched refs:LaunchDimensions (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dpartition_assignment.h35 class LaunchDimensions {
39 LaunchDimensions() : block_count_(1), threads_per_block_(1) {} in LaunchDimensions() function
41 LaunchDimensions(int64 block_count, int64 threads_per_block) in LaunchDimensions() function
58 const LaunchDimensions& launch_dims);
63 LaunchDimensions CalculateLaunchDimensions(
Dparallel_loop_emitter.h36 const LaunchDimensions& launch_dimensions,
42 const LaunchDimensions& launch_dimensions,
52 const LaunchDimensions& launch_dimensions,
64 const LaunchDimensions launch_dimensions_;
Dpartition_assignment.cc36 const LaunchDimensions& launch_dims) { in operator <<()
62 LaunchDimensions CalculateLaunchDimensions( in CalculateLaunchDimensions()
67 return LaunchDimensions(); in CalculateLaunchDimensions()
97 return LaunchDimensions(block_count, threads_per_block); in CalculateLaunchDimensions()
Dparallel_loop_emitter.cc34 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
43 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
52 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
Dkernel_thunk.h59 void SetLaunchDimensions(const LaunchDimensions& launch_dims);
82 LaunchDimensions launch_dimensions_;
Dkernel_thunk.cc69 void KernelThunk::SetLaunchDimensions(const LaunchDimensions& launch_dims) { in SetLaunchDimensions()
79 LaunchDimensions launch_dimensions; in ExecuteOnStream()
Dir_emitter_unnested.h243 LaunchDimensions EmitHlo021Tile(HloInstruction* hlo,
247 LaunchDimensions EmitKernel(HloInstruction* unnested_hlo,
Dir_emitter_unnested.cc147 void UpdateLaunchDimensions(const LaunchDimensions& launch_dims, Thunk* thunk, in UpdateLaunchDimensions()
664 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleFusion()
807 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleSelectAndScatter()
1193 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitScatter()
1278 LaunchDimensions standard_launch_dimensions = CalculateLaunchDimensions( in HandleSort()
1316 LaunchDimensions tiled_launch_dimensions(num_blocks, kThreadsPerBlock); in HandleSort()
1321 LaunchDimensions launch_dimensions = xor_masks.size() > 1 in HandleSort()
1991 LaunchDimensions launch_dimensions = in BuildInitializerThunk()
2186 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitTargetElementLoopInThunk()
3050 LaunchDimensions IrEmitterUnnested::EmitKernel( in EmitKernel()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc44 const IrArray& output_array, const gpu::LaunchDimensions* launch_dimensions, in EmitDynamicUpdateSliceInPlaceImpl()
141 const gpu::LaunchDimensions* launch_dimensions, llvm::IRBuilder<>* b) { in EmitFusedDynamicUpdateSliceInPlaceImpl()
200 const gpu::LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b) { in EmitParallelFusedDynamicUpdateSliceInPlace()
Dsort_util.h40 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions,
Ddynamic_update_slice_util.h90 const gpu::LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b);
Dsort_util.cc295 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions, in EmitSortInPlace()