Searched refs:LoopEmitter (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.h | 39 class LoopEmitter {
43 LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape,
47 LoopEmitter(const ElementGenerator& target_element_generator,
55 LoopEmitter(const ElementGenerator& target_element_generator,
58 LoopEmitter(const LoopEmitter&) = delete;
59 LoopEmitter& operator=(const LoopEmitter&) = delete;
60 virtual ~LoopEmitter() = default;
|
| D | loop_emitter.cc | 34 LoopEmitter::LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape, in LoopEmitter() function in xla::llvm_ir::LoopEmitter
38 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter() function in xla::llvm_ir::LoopEmitter
50 static LoopEmitter::BodyEmitter MakeBodyEmitterForMultiOutputFusion( in MakeBodyEmitterForMultiOutputFusion()
70 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter() function in xla::llvm_ir::LoopEmitter
87 std::vector<IrArray::Index> LoopEmitter::EmitIndexAndSetExitBasicBlock( in EmitIndexAndSetExitBasicBlock()
125 Status LoopEmitter::EmitLoop(absl::string_view loop_name, in EmitLoop()
|
| D | dynamic_update_slice_util.cc | 103 return LoopEmitter(loop_body_emitter, update_shape, b).EmitLoop(name); in EmitDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 36 : LoopEmitter(body_emitter, shape, b), in ParallelLoopEmitter()
45 : LoopEmitter(target_element_generator, target_arrays, b), in ParallelLoopEmitter()
54 : LoopEmitter(target_element_generator, target_array, b), in ParallelLoopEmitter()
|
| D | ir_emitter_nested.cc | 125 llvm_ir::LoopEmitter(element_generator, target_arrays, &b_).EmitLoop()); in EmitTargetElementLoop()
129 return llvm_ir::LoopEmitter(element_generator, GetIrArray(hlo, hlo), &b_) in EmitTargetElementLoop()
|
| D | parallel_loop_emitter.h | 31 class ParallelLoopEmitter : public llvm_ir::LoopEmitter {
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.h | 48 class ParallelLoopEmitter : public llvm_ir::LoopEmitter {
|
| D | parallel_loop_emitter.cc | 29 : LoopEmitter(target_element_generator, target_array, b), in ParallelLoopEmitter()
|
| D | ir_emitter.cc | 2970 llvm_ir::LoopEmitter(element_generator, output_arrays, &b_) in EmitTargetElementLoop()
2990 llvm_ir::LoopEmitter(element_generator, target_array, &b_) in EmitTargetElementLoop()
|