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.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.cc79 LaunchDimensions launch_dimensions; in ExecuteOnStream() local
Dir_emitter_unnested.cc664 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleFusion() local
807 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleSelectAndScatter() local
1193 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitScatter() local
1321 LaunchDimensions launch_dimensions = xor_masks.size() > 1 in HandleSort() local
1991 LaunchDimensions launch_dimensions = in BuildInitializerThunk() local
2186 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitTargetElementLoopInThunk() local
3076 LaunchDimensions launch_dimensions = in EmitKernel() local
3453 const LaunchDimensions launch_dimensions = in CheckAndEmitHloWithTile021() local
3793 LaunchDimensions launch_dimensions = in EmitReductionToVector() local
/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.cc295 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions, in EmitSortInPlace()