Searched defs:launch_dimensions (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | dynamic_update_slice_util.cc | 98 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()
|
| H A D | sort_util.cc | 321 llvm::IRBuilder<>* b, const gpu::LaunchDimensions& launch_dimensions, in EmitSortInPlace()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | kernel_thunk.cc | 39 const LaunchDimensions& launch_dimensions) in KernelThunk()
92 LaunchDimensions launch_dimensions; in ExecuteOnStream() local
|
| H A D | parallel_loop_emitter.cc | 36 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
47 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* b, in ParallelLoopEmitter()
|
| H A D | ir_emitter_unnested.cc | 1548 LaunchDimensionsConfig launch_config, LaunchDimensions launch_dimensions, in EnableLogicalIndexGenerationForOutput()
1622 LaunchDimensions launch_dimensions( in EmitLaunchFunc() local
2280 const LaunchDimensions& launch_dimensions, const llvm_ir::IrArray& output, in EmitScatter()
2309 const LaunchDimensions& launch_dimensions) { in EmitScatter()
2692 LaunchDimensions launch_dimensions = xor_masks.size() > 1 in EmitSort() local
2960 const LaunchDimensions& launch_dimensions) { in BuildKernelThunkImpl()
3075 const LaunchDimensions& launch_dimensions) { in BuildKernelThunk()
3096 const LaunchDimensions& launch_dimensions) { in BuildKernelThunk()
4206 const LaunchDimensions& launch_dimensions) { in EmitHlo021Tile()
4325 LaunchDimensions launch_dimensions( in Emit021Transpose() local
[all …]
|
| H A D | kernel_thunk.h | 66 const LaunchDimensions& launch_dimensions() const { in launch_dimensions() function
|
| H A D | jitrt_custom_calls.cc | 429 LaunchDimensions launch_dimensions( in operator ()() local
|