Searched refs:body_emitter (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 33 BodyEmitter body_emitter, const Shape& shape, in ParallelLoopEmitter() argument
36 : LoopEmitter(body_emitter, shape, b), in ParallelLoopEmitter()
|
| 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_unnested.h | 186 const llvm_ir::ElementGenerator& body_emitter) override;
191 const HloInstruction& hlo, const llvm_ir::ElementGenerator& body_emitter,
|
| D | ir_emitter.h | 151 const llvm_ir::ElementGenerator& body_emitter) = 0;
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 34 LoopEmitter::LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape, in LoopEmitter() argument
36 : body_emitter_(body_emitter), shape_(shape), b_(b) {} in LoopEmitter()
|
| D | loop_emitter.h | 43 LoopEmitter(const BodyEmitter& body_emitter, const Shape& shape,
|