Home
last modified time | relevance | path

Searched refs:ParallelLoopEmitter (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dparallel_loop_emitter.h48 class ParallelLoopEmitter : public llvm_ir::LoopEmitter {
54 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
59 ParallelLoopEmitter(const ParallelLoopEmitter&) = delete;
60 ParallelLoopEmitter& operator=(const ParallelLoopEmitter&) = delete;
61 ~ParallelLoopEmitter() override = default;
H A Dparallel_loop_emitter.cc25 ParallelLoopEmitter::ParallelLoopEmitter( in ParallelLoopEmitter() function in xla::cpu::ParallelLoopEmitter
33 ParallelLoopEmitter::EmitIndexAndSetExitBasicBlock(absl::string_view loop_name, in EmitIndexAndSetExitBasicBlock()
H A Dir_emitter.cc3373 TF_RETURN_IF_ERROR(ParallelLoopEmitter(element_generator, target_array, in EmitTargetElementLoop()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dparallel_loop_emitter.h31 class ParallelLoopEmitter {
36 ParallelLoopEmitter(llvm_ir::BodyEmitter body_emitter, const Shape& shape,
46 ParallelLoopEmitter(const llvm_ir::ElementGenerator& target_element_generator,
52 ParallelLoopEmitter(const ParallelLoopEmitter&) = delete;
53 ParallelLoopEmitter& operator=(const ParallelLoopEmitter&) = delete;
H A Dparallel_loop_emitter.cc34 ParallelLoopEmitter::ParallelLoopEmitter( in ParallelLoopEmitter() function in xla::gpu::ParallelLoopEmitter
44 ParallelLoopEmitter::ParallelLoopEmitter( in ParallelLoopEmitter() function in xla::gpu::ParallelLoopEmitter
58 ParallelLoopEmitter::LinearBaseAndThreadIdx
59 ParallelLoopEmitter::EmitLinearBaseAndThreadIdx(llvm::Type* index_type, in EmitLinearBaseAndThreadIdx()
147 ParallelLoopEmitter::EmitLogicalIndexAndSetExitBasicBlock( in EmitLogicalIndexAndSetExitBasicBlock()
202 ParallelLoopEmitter::EmitIndexAndSetExitBasicBlock(absl::string_view loop_name, in EmitIndexAndSetExitBasicBlock()
298 Status ParallelLoopEmitter::EmitSerialLoop(absl::string_view loop_name, in EmitSerialLoop()
308 Status ParallelLoopEmitter::EmitLoop(absl::string_view loop_name, in EmitLoop()
H A Dir_emitter_unnested.cc843 TF_RETURN_IF_ERROR(ParallelLoopEmitter(body_generator, data_shape, in EmitPadToStatic()
966 TF_RETURN_IF_ERROR(ParallelLoopEmitter(body_generator, data_shape, in EmitSliceToDynamic()
1776 ParallelLoopEmitter(element_generator, output_element_arrays, in EmitLoopFusion()
2145 return ParallelLoopEmitter(loop_body_emitter, source_shape, launch_dimensions, in EmitSelectAndScatter()
2436 return ParallelLoopEmitter(loop_body_emitter, desc.updates_shape, in EmitScatter()
3233 TF_RETURN_IF_ERROR(ParallelLoopEmitter( in BuildInitializerThunk()
3299 ParallelLoopEmitter(generator, {dest_array}, launch_dimensions, &b_) in BuildFusedInitializerThunk()
5133 ParallelLoopEmitter( in EmitInputFusibleNonStridedSlices()
5234 ParallelLoopEmitter(generator, {ir_arrays.back()}, launch_dimensions, in EmitScatter()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Ddynamic_update_slice_util.cc153 return gpu::ParallelLoopEmitter(loop_body_emitter, update_shape, in EmitDynamicUpdateSliceInPlaceImpl()
H A Dsort_util.cc420 return gpu::ParallelLoopEmitter(compare_loop_body_emitter, iteration_shape, in EmitSortInPlace()