Home
last modified time | relevance | path

Searched defs:launch_dimensions (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc35 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
44 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
53 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
Dkernel_thunk.cc75 LaunchDimensions launch_dimensions; in ExecuteOnStream() local
Dir_emitter_unnested.cc482 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleFusion() local
628 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleSelectAndScatter() local
1014 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitScatter() local
1142 LaunchDimensions launch_dimensions = xor_masks.size() > 1 in HandleSort() local
1608 LaunchDimensions launch_dimensions = in BuildInitializerThunk() local
1804 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitTargetElementLoopInThunk() local
2540 LaunchDimensions launch_dimensions(mapping_scheme.GetNumberOfBlocks(), in EmitHlo021Tile() local
3049 LaunchDimensions launch_dimensions(mapping_scheme.GetNumberOfBlocks(), in EmitReductionFromOrToContiguousDimensions() local
3227 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitInputFusibleNonStridedSlices() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc98 const IrArray& output_array, const gpu::LaunchDimensions* launch_dimensions, in EmitDynamicUpdateSliceInPlaceImpl()
195 const gpu::LaunchDimensions* launch_dimensions, llvm::IRBuilder<>* b) { in EmitFusedDynamicUpdateSliceInPlaceImpl()
254 const gpu::LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b) { in EmitParallelFusedDynamicUpdateSliceInPlace()
Dsort_util.cc297 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions, in EmitSortInPlace()