Home
last modified time | relevance | path

Searched refs:IrEmitter (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc57 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config, in IrEmitter() function in xla::gpu::IrEmitter
68 Status IrEmitter::DefaultAction(HloInstruction* hlo) { in DefaultAction()
81 Status IrEmitter::HandleConstant(HloInstruction* constant) { in HandleConstant()
85 Status IrEmitter::HandleBitcast(HloInstruction* bitcast) { in HandleBitcast()
97 Status IrEmitter::HandleAddDependency(HloInstruction* add_dependency) { in HandleAddDependency()
109 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement()
122 Status IrEmitter::HandleSend(HloInstruction*) { in HandleSend()
126 Status IrEmitter::HandleSendDone(HloInstruction*) { in HandleSendDone()
130 Status IrEmitter::HandleRecv(HloInstruction*) { in HandleRecv()
134 Status IrEmitter::HandleRecvDone(HloInstruction*) { in HandleRecvDone()
[all …]
Dir_emitter.h68 class IrEmitter : public DfsHloVisitorWithDefault,
69 public IrBuilderMixin<IrEmitter> {
74 IrEmitter(const IrEmitter&) = delete;
75 IrEmitter& operator=(const IrEmitter&) = delete;
113 explicit IrEmitter(const HloModuleConfig& hlo_module_config,
169 return std::bind(&IrEmitter::ComputeNestedElement, this, in GetNestedComputer()
Dir_emitter_nested.h40 class IrEmitterNested : public IrEmitter {
Dir_emitter_nested.cc41 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/true) { in IrEmitterNested()
Dir_emitter_unnested.cc177 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/false), in IrEmitterUnnested()
349 return IrEmitter::DefaultAction(hlo); in DefaultAction()
359 return IrEmitter::HandleDot(dot); in HandleDot()
370 return IrEmitter::HandleConvolution(convolution); in HandleConvolution()
529 return IrEmitter::HandleCustomCall(custom_call); in HandleCustomCall()
686 return IrEmitter::HandleFusion(fusion); in HandleFusion()
703 return IrEmitter::HandleCopy(copy); in HandleCopy()
731 return IrEmitter::HandleReduce(reduce); in HandleReduce()
772 return IrEmitter::HandleTuple(tuple); in HandleTuple()
1206 return IrEmitter::HandleSelect(select); in HandleSelect()
[all …]
Dir_emitter_unnested.h50 class IrEmitterUnnested : public IrEmitter {
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc83 IrEmitter::IrEmitter( in IrEmitter() function in xla::cpu::IrEmitter
109 StatusOr<llvm::Function*> IrEmitter::EmitComputation( in EmitComputation()
152 void IrEmitter::InitializeIrFunction(const string& function_name) { in InitializeIrFunction()
165 IrEmitter::~IrEmitter() {} in ~IrEmitter()
167 Status IrEmitter::HandleBitcast(HloInstruction* bitcast) { in HandleBitcast()
175 llvm::Constant* IrEmitter::EmitGlobalForLiteral(const Literal& literal) { in EmitGlobalForLiteral()
191 Status IrEmitter::EmitConstantGlobals() { in EmitConstantGlobals()
214 Status IrEmitter::HandleConstant(HloInstruction* constant) { in HandleConstant()
221 Status IrEmitter::HandleCopy(HloInstruction* copy) { in HandleCopy()
235 int IrEmitter::MinimumAlignmentForPrimitiveType(PrimitiveType primitive_type) { in MinimumAlignmentForPrimitiveType()
[all …]
Delemental_ir_emitter.h33 IrEmitter* ir_emitter, llvm::Module* module) in CpuElementalIrEmitter()
47 IrEmitter* ir_emitter_;
Dir_emitter.h59 class IrEmitter : public DfsHloVisitorWithDefault,
60 public IrBuilderMixin<IrEmitter> {
76 IrEmitter(const HloModule& hlo_module, const BufferAssignment& assignment,
84 ~IrEmitter() override;
588 TF_DISALLOW_COPY_AND_ASSIGN(IrEmitter);
Dcpu_compiler.cc664 IrEmitter ir_emitter(*module, *assignment, llvm_module.get(), in RunBackend()
855 IrEmitter ir_emitter(*module, *assignment, &llvm_module, in CompileAheadOfTime()