| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.h | 40 class LoopEmitter {
44 LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape,
49 LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape,
54 LoopEmitter(const ElementGenerator& target_element_generator,
62 LoopEmitter(const ElementGenerator& target_element_generator,
65 LoopEmitter(const LoopEmitter&) = delete;
66 LoopEmitter& operator=(const LoopEmitter&) = delete;
67 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 BodyEmitter& body_emitter, const Shape& shape, in LoopEmitter() function in xla::llvm_ir::LoopEmitter
41 : LoopEmitter::LoopEmitter(body_emitter, shape, b) { in LoopEmitter()
46 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter() function in xla::llvm_ir::LoopEmitter
58 static LoopEmitter::BodyEmitter MakeBodyEmitterForMultiOutput( in MakeBodyEmitterForMultiOutput()
78 LoopEmitter::LoopEmitter(const ElementGenerator& target_element_generator, in LoopEmitter() function in xla::llvm_ir::LoopEmitter
95 IrArray::Index LoopEmitter::EmitStaticIndex(ForLoopNest* loop_nest, in EmitStaticIndex()
113 IrArray::Index LoopEmitter::EmitDynamicIndex(ForLoopNest* loop_nest, in EmitDynamicIndex()
132 std::vector<IrArray::Index> LoopEmitter::EmitIndexAndSetExitBasicBlock( in EmitIndexAndSetExitBasicBlock()
166 Status LoopEmitter::EmitLoop(absl::string_view loop_name, in EmitLoop()
|
| D | dynamic_update_slice_util.cc | 157 return LoopEmitter(loop_body_emitter, update_shape, b).EmitLoop(name); in EmitDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 38 : LoopEmitter(body_emitter, shape, b), in ParallelLoopEmitter()
47 : LoopEmitter(target_element_generator, target_arrays, b), in ParallelLoopEmitter()
56 : LoopEmitter(target_element_generator, target_array, b), in ParallelLoopEmitter()
169 return LoopEmitter::EmitLoop(loop_name, index_type); in EmitLoop()
|
| D | ir_emitter_nested.cc | 175 llvm_ir::LoopEmitter(element_generator, target_arrays, &b_).EmitLoop()); in EmitTargetElementLoop()
179 return llvm_ir::LoopEmitter(element_generator, GetIrArray(hlo, hlo), &b_) in EmitTargetElementLoop()
|
| D | parallel_loop_emitter.h | 31 class ParallelLoopEmitter : public llvm_ir::LoopEmitter {
|
| D | ir_emitter_unnested.cc | 906 llvm_ir::LoopEmitter::BodyEmitter body_generator = in EmitPadToStaticFromMlir()
1027 llvm_ir::LoopEmitter::BodyEmitter body_generator = in EmitSliceToDynamicFromMlir()
|
| /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 | 2183 return llvm_ir::LoopEmitter(loop_body_emitter, data_array.GetShape(), in HandleSliceToDynamic()
2248 llvm_ir::LoopEmitter(loop_body_emitter, input_shape, dynamic_dims, &b_) in HandlePadToStatic()
3154 llvm_ir::LoopEmitter(element_generator, output_arrays, &b_) in EmitTargetElementLoop()
3174 llvm_ir::LoopEmitter(element_generator, target_array, &b_) in EmitTargetElementLoop()
|