Searched refs:IrEmitterUnnested (Results 1 – 3 of 3) sorted by relevance
531 IrEmitterUnnested::IrEmitterUnnested(const HloModuleConfig& hlo_module_config, in IrEmitterUnnested() function in xla::gpu::IrEmitterUnnested536 StatusOr<std::unique_ptr<IrEmitterUnnested>> IrEmitterUnnested::Create( in Create()540 auto emitter = std::unique_ptr<IrEmitterUnnested>(new IrEmitterUnnested( in Create()553 Status IrEmitterUnnested::Postprocess(HloInstruction* hlo) { in Postprocess()558 llvm::Function* IrEmitterUnnested::BuildKernelPrototype( in BuildKernelPrototype()623 StatusOr<BufferAllocation::Slice> IrEmitterUnnested::GetAllocationSliceForMlir( in GetAllocationSliceForMlir()629 Status IrEmitterUnnested::DefaultAction(HloInstruction* hlo) { in DefaultAction()634 Status IrEmitterUnnested::EmitUsingElementalIrEmitter(MlirEmitterInput input) { in EmitUsingElementalIrEmitter()736 Status IrEmitterUnnested::HandleConstant(HloInstruction* constant) { in HandleConstant()741 Status IrEmitterUnnested::EmitConstant(MlirEmitterInput mlir_input) { in EmitConstant()[all …]
106 class IrEmitterUnnested : public IrEmitter,146 IrEmitterUnnested(const IrEmitterUnnested&) = delete;147 IrEmitterUnnested& operator=(const IrEmitterUnnested&) = delete;149 static StatusOr<std::unique_ptr<IrEmitterUnnested>> Create(240 IrEmitterUnnested(const HloModuleConfig& hlo_module_config,553 const IrEmitterUnnested::EmitElementFunction& emit_elem_function);
605 IrEmitterUnnested::Create(hlo_module->config(), entry_computation, in CompileModuleToLlvmIrImpl()1119 IrEmitterUnnested::Create(module_config, /*hlo_computation=*/nullptr, in CompileLmhloToExecutable()