Searched refs:ParallelLoopEmitter (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.h | 48 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;
|
| D | parallel_loop_emitter.cc | 25 ParallelLoopEmitter::ParallelLoopEmitter( in ParallelLoopEmitter() function in xla::cpu::ParallelLoopEmitter
33 ParallelLoopEmitter::EmitIndexAndSetExitBasicBlock(absl::string_view loop_name, in EmitIndexAndSetExitBasicBlock()
|
| D | ir_emitter.cc | 3373 TF_RETURN_IF_ERROR(ParallelLoopEmitter(element_generator, target_array, in EmitTargetElementLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.h | 31 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;
|
| D | parallel_loop_emitter.cc | 34 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()
|
| D | ir_emitter_unnested.cc | 843 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 153 return gpu::ParallelLoopEmitter(loop_body_emitter, update_shape, in EmitDynamicUpdateSliceInPlaceImpl()
|
| D | sort_util.cc | 420 return gpu::ParallelLoopEmitter(compare_loop_body_emitter, iteration_shape, in EmitSortInPlace()
|