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.cc36 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
45 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
54 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
Dkernel_thunk.cc90 LaunchDimensions launch_dimensions; in ExecuteOnStream() local
Dir_emitter_unnested.cc924 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitPadToStaticFromMlir() local
1047 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitSliceToDynamicFromMlir() local
1841 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitLoopFusionFromMlir() local
1918 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleFusion() local
2042 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in HandleFusion() local
2238 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitSelectAndScatterFromMlir() local
2722 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in EmitScatter() local
2994 LaunchDimensions launch_dimensions = xor_masks.size() > 1 in EmitSortFromMlir() local
3673 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in BuildInitializerThunkForMlir() local
3717 LaunchDimensions launch_dimensions = CalculateLaunchDimensions( in BuildFusedInitializerThunkForMlir() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc98 const IrArray& output_array, const gpu::LaunchDimensions* launch_dimensions, in EmitDynamicUpdateSliceInPlaceImpl()
194 const gpu::LaunchDimensions* launch_dimensions, llvm::IRBuilder<>* b) { in EmitFusedDynamicUpdateSliceInPlaceImpl()
248 const gpu::LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b) { in EmitParallelFusedDynamicUpdateSliceInPlace()
Dsort_util.cc297 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions, in EmitSortInPlace()