Home
last modified time | relevance | path

Searched refs:ir_emitter (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.h33 IrEmitter* ir_emitter, llvm::Module* module) in CpuElementalIrEmitter() argument
34 : ElementalIrEmitter(module, ir_emitter->b()), in CpuElementalIrEmitter()
36 ir_emitter_(ir_emitter) {} in CpuElementalIrEmitter()
Dcpu_compiler.cc737 IrEmitter ir_emitter(&mlir_context, *module, *assignment, llvm_module.get(), in RunBackend() local
748 TF_RETURN_IF_ERROR(ir_emitter.EmitConstantGlobals()); in RunBackend()
756 ir_emitter in RunBackend()
767 ir_emitter.EmitComputation( in RunBackend()
945 IrEmitter ir_emitter(&mlir_context, *module, *assignment, &llvm_module, in CompileAheadOfTime() local
952 TF_RETURN_IF_ERROR(ir_emitter.EmitConstantGlobals()); in CompileAheadOfTime()
961 ir_emitter in CompileAheadOfTime()
970 ir_emitter.EmitComputation( in CompileAheadOfTime()
DBUILD144 ":ir_emitter",
363 name = "ir_emitter",
366 "ir_emitter.cc",
370 "ir_emitter.h",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc604 auto ir_emitter, in CompileModuleToLlvmIrImpl()
615 TF_RETURN_IF_ERROR(instruction->Visit(ir_emitter.get())); in CompileModuleToLlvmIrImpl()
616 TF_RETURN_IF_ERROR(ir_emitter->Postprocess(instruction)); in CompileModuleToLlvmIrImpl()
618 ir_emitter->ConsumeThunkSequence(); in CompileModuleToLlvmIrImpl()
1118 auto ir_emitter, in CompileLmhloToExecutable()
1126 TF_RETURN_IF_ERROR(ir_emitter->EmitOp(input)); in CompileLmhloToExecutable()
1127 std::unique_ptr<ThunkSequence> thunks = ir_emitter->ConsumeThunkSequence(); in CompileLmhloToExecutable()
DBUILD235 name = "ir_emitter",
237 "ir_emitter.cc",
242 "ir_emitter.h",
1256 ":ir_emitter",
Dir_emitter_unnested.cc3914 auto ir_emitter, in BuildConditionalThunk()
3917 TF_CHECK_OK(branch_computation->Accept(ir_emitter.get())); in BuildConditionalThunk()
3918 branch_thunks.push_back(std::move(*ir_emitter->ConsumeThunkSequence())); in BuildConditionalThunk()