Home
last modified time | relevance | path

Searched refs:ElementalIrEmitter (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc236 StatusOr<llvm::Value*> ElementalIrEmitter::EmitUnaryOp( in EmitUnaryOp()
248 StatusOr<llvm::Value*> ElementalIrEmitter::EmitIntegerUnaryOp( in EmitIntegerUnaryOp()
368 StatusOr<llvm::Value*> ElementalIrEmitter::EmitFloatUnaryOp( in EmitFloatUnaryOp()
519 StatusOr<llvm::Value*> ElementalIrEmitter::EmitComplexUnaryOp( in EmitComplexUnaryOp()
814 StatusOr<llvm::Value*> ElementalIrEmitter::EmitBinaryOp( in EmitBinaryOp()
829 StatusOr<llvm::Value*> ElementalIrEmitter::EmitFloatBinaryOp( in EmitFloatBinaryOp()
901 ElementalIrEmitter::EmitComplexAbsHelper(PrimitiveType prim_type, in EmitComplexAbsHelper()
921 StatusOr<llvm::Value*> ElementalIrEmitter::EmitComplexAbs( in EmitComplexAbs()
937 StatusOr<llvm::Value*> ElementalIrEmitter::EmitSqrtComplexAbs( in EmitSqrtComplexAbs()
957 StatusOr<llvm::Value*> ElementalIrEmitter::EmitRsqrtComplexAbs( in EmitRsqrtComplexAbs()
[all …]
Delemental_ir_emitter.h36 class ElementalIrEmitter : public IrBuilderMixin<ElementalIrEmitter> {
41 ElementalIrEmitter(llvm::Module* module, llvm::IRBuilder<>* b) in ElementalIrEmitter() function
44 virtual ~ElementalIrEmitter() = default;
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.h30 class CpuElementalIrEmitter : public ElementalIrEmitter {
34 : ElementalIrEmitter(module, ir_emitter->b()), in CpuElementalIrEmitter()
Dir_emitter.cc3212 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.h57 explicit FusedIrEmitter(ElementalIrEmitter* elemental_emitter) in FusedIrEmitter()
90 ElementalIrEmitter* elemental_emitter_;
/external/tensorflow/tensorflow/compiler/mlir/g3doc/
Dxla_gpu_codegen.md94 of "math" ops are implemented directly in (Cpu|Gpu)ElementalIrEmitter.
96 ElementalIrEmitter is unique in that:
101 * Most fusions solely depend on ElementalIrEmitter.
141 * ElementalIrEmitter ops go for (4), but not incrementally. There is no way to
238 ### Step 4: (Task 3) Migrated ElementalIrEmitter
240 Once profiling is ready, we can complete and tune all ElementalIrEmitter-based
244 Notice that it's beneficial to migrate XLA/CPU's ElementalIrEmitter as well,
249 ElementalIrEmitter.
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc75 : ElementalIrEmitter(module, b), in GpuElementalIrEmitter()
187 return ElementalIrEmitter::EmitFloatBinaryOp(op, lhs_value, rhs_value); in EmitFloatBinaryOp()
Delemental_ir_emitter.h39 class GpuElementalIrEmitter : public ElementalIrEmitter {
Dir_emitter.cc91 ElementalIrEmitter::HloToElementGeneratorMap operand_to_generator; in DefaultAction()