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.cc38 const LaunchDimensions& launch_dimensions) in KernelThunk()
90 LaunchDimensions launch_dimensions; in ExecuteOnStream() local
Dir_emitter_unnested.cc2420 const LaunchDimensions& launch_dimensions, const llvm_ir::IrArray& output, in EmitScatter()
2449 const LaunchDimensions& launch_dimensions) { in EmitScatter()
2837 LaunchDimensions launch_dimensions = xor_masks.size() > 1 in EmitSort() local
3096 const LaunchDimensions& launch_dimensions) { in BuildKernelThunkImpl()
3209 const LaunchDimensions& launch_dimensions) { in BuildKernelThunk()
3229 const LaunchDimensions& launch_dimensions) { in BuildKernelThunk()
4374 const LaunchDimensions& launch_dimensions) { in EmitHlo021Tile()
4698 LaunchDimensions launch_dimensions(mapping_scheme.GetNumberOfBlocks(), in CheckAndEmitHloWithTile021() local
4951 LaunchDimensions launch_dimensions(mapping_scheme.GetNumberOfBlocks(), in EmitIRForReduction() local
5111 LaunchDimensions launch_dimensions( in EmitUnnestedReduction() 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()
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()