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.cc79 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config, in IrEmitter() function in xla::gpu::IrEmitter
90 Status IrEmitter::DefaultAction(HloInstruction* hlo) { in DefaultAction()
104 Status IrEmitter::EmitConstants(const HloComputation& computation, in EmitConstants()
166 Status IrEmitter::HandleConstant(HloInstruction* constant) { in HandleConstant()
170 Status IrEmitter::HandleAddDependency(HloInstruction* add_dependency) { in HandleAddDependency()
182 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement()
195 Status IrEmitter::HandleSend(HloInstruction*) { in HandleSend()
199 Status IrEmitter::HandleSendDone(HloInstruction*) { in HandleSendDone()
203 Status IrEmitter::HandleRecv(HloInstruction*) { in HandleRecv()
207 Status IrEmitter::HandleRecvDone(HloInstruction*) { in HandleRecvDone()
[all …]
Dir_emitter.h69 class IrEmitter : public DfsHloVisitorWithDefault,
70 public IrBuilderMixin<IrEmitter> {
75 IrEmitter(const IrEmitter&) = delete;
76 IrEmitter& operator=(const IrEmitter&) = delete;
116 explicit IrEmitter(const HloModuleConfig& hlo_module_config,
165 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.h106 class IrEmitterUnnested : public IrEmitter,
Dir_emitter_unnested.cc534 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/false) {} in IrEmitterUnnested()
809 return IrEmitter::HandleConvolution(convolution); in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc91 IrEmitter::IrEmitter( in IrEmitter() function in xla::cpu::IrEmitter
118 void IrEmitter::EmitThreadLocalFunctionEpilogue(HloComputation* computation) { in EmitThreadLocalFunctionEpilogue()
154 StatusOr<llvm::Function*> IrEmitter::EmitComputation( in EmitComputation()
211 void IrEmitter::InitializeIrFunction(const string& function_name) { in InitializeIrFunction()
224 IrEmitter::~IrEmitter() {} in ~IrEmitter()
226 Status IrEmitter::HandleBitcast(HloInstruction* bitcast) { in HandleBitcast()
234 llvm::Constant* IrEmitter::EmitGlobalForLiteral(const Literal& literal) { in EmitGlobalForLiteral()
251 Status IrEmitter::EmitConstantGlobals() { in EmitConstantGlobals()
274 Status IrEmitter::HandleConstant(HloInstruction* constant) { in HandleConstant()
281 Status IrEmitter::HandleCopy(HloInstruction* copy) { in HandleCopy()
[all …]
Delemental_ir_emitter.h33 IrEmitter* ir_emitter, llvm::Module* module) in CpuElementalIrEmitter()
56 IrEmitter* ir_emitter_;
Dir_emitter.h63 class IrEmitter : public DfsHloVisitorWithDefault,
64 public IrBuilderMixin<IrEmitter> {
84 IrEmitter(mlir::MLIRContext* mlir_context, const HloModule& hlo_module,
92 ~IrEmitter() override;
614 TF_DISALLOW_COPY_AND_ASSIGN(IrEmitter);
Dcpu_compiler.cc737 IrEmitter ir_emitter(&mlir_context, *module, *assignment, llvm_module.get(), in RunBackend()
945 IrEmitter ir_emitter(&mlir_context, *module, *assignment, &llvm_module, in CompileAheadOfTime()