Searched refs:body_emitter (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 34 BodyEmitter body_emitter, const Shape& shape, in ParallelLoopEmitter() argument
36 : LoopEmitter(body_emitter, shape, ir_builder), in ParallelLoopEmitter()
|
| D | ir_emitter_unnested.h | 82 const llvm_ir::ElementGenerator& body_emitter) override;
87 const HloInstruction& hlo, const llvm_ir::ElementGenerator& body_emitter,
|
| D | ir_emitter_nested.h | 59 const llvm_ir::ElementGenerator& body_emitter) override;
|
| D | parallel_loop_emitter.h | 35 ParallelLoopEmitter(BodyEmitter body_emitter, const Shape& shape,
|
| D | ir_emitter.h | 136 const llvm_ir::ElementGenerator& body_emitter) = 0;
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 35 LoopEmitter::LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape, in LoopEmitter() argument
37 : body_emitter_(body_emitter), shape_(shape), ir_builder_(ir_builder) {} in LoopEmitter()
|
| D | loop_emitter.h | 44 LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape,
|