Searched refs:launch_dimensions (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 34 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter() argument
37 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
43 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter() argument
46 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
52 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter() argument
55 launch_dimensions_(launch_dimensions), in ParallelLoopEmitter()
|
| D | kernel_thunk.cc | 79 LaunchDimensions launch_dimensions; in ExecuteOnStream() local
87 launch_dimensions = launch_dimensions_; in ExecuteOnStream()
103 stream, se::ThreadDim(launch_dimensions.threads_per_block()), in ExecuteOnStream()
104 se::BlockDim(launch_dimensions.block_count()), *kernel, in ExecuteOnStream()
|
| D | parallel_loop_emitter.h | 36 const LaunchDimensions& launch_dimensions,
42 const LaunchDimensions& launch_dimensions,
52 const LaunchDimensions& launch_dimensions,
|
| D | ir_emitter_unnested.cc | 664 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleFusion() local
666 UpdateLaunchDimensions(launch_dimensions, fusion_thunk.get(), in HandleFusion()
672 &elemental_emitter, launch_dimensions, &b_); in HandleFusion()
807 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleSelectAndScatter() local
810 select_and_scatter, launch_dimensions.launch_bound(), &b_); in HandleSelectAndScatter()
964 launch_dimensions, in HandleSelectAndScatter()
973 launch_dimensions, &b_) in HandleSelectAndScatter()
1193 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitScatter() local
1195 UpdateLaunchDimensions(launch_dimensions, thunk, in EmitScatter()
1199 launch_dimensions, &b_) in EmitScatter()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 44 const IrArray& output_array, const gpu::LaunchDimensions* launch_dimensions, in EmitDynamicUpdateSliceInPlaceImpl() argument
98 if (launch_dimensions != nullptr) { in EmitDynamicUpdateSliceInPlaceImpl()
100 *launch_dimensions, b) in EmitDynamicUpdateSliceInPlaceImpl()
141 const gpu::LaunchDimensions* launch_dimensions, llvm::IRBuilder<>* b) { in EmitFusedDynamicUpdateSliceInPlaceImpl() argument
182 fusion_output_array, launch_dimensions, IrName(fusion), b); in EmitFusedDynamicUpdateSliceInPlaceImpl()
200 const gpu::LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b) { in EmitParallelFusedDynamicUpdateSliceInPlace() argument
203 elemental_emitter, &launch_dimensions, b); in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| D | sort_util.h | 40 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions,
|
| D | dynamic_update_slice_util.h | 90 const gpu::LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b);
|
| D | sort_util.cc | 295 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions, in EmitSortInPlace() argument
390 launch_dimensions, b) in EmitSortInPlace()
|