Searched refs:ir_emitter (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | elemental_ir_emitter.h | 33 IrEmitter* ir_emitter, llvm::Module* module) in CpuElementalIrEmitter() argument
34 : ElementalIrEmitter(module_config, module, ir_emitter->b()), in CpuElementalIrEmitter()
35 ir_emitter_(ir_emitter) {} in CpuElementalIrEmitter()
|
| D | cpu_compiler.cc | 664 IrEmitter ir_emitter(*module, *assignment, llvm_module.get(), in RunBackend() local
675 TF_RETURN_IF_ERROR(ir_emitter.EmitConstantGlobals()); in RunBackend()
683 ir_emitter in RunBackend()
694 ir_emitter.EmitComputation( in RunBackend()
855 IrEmitter ir_emitter(*module, *assignment, &llvm_module, in CompileAheadOfTime() local
862 TF_RETURN_IF_ERROR(ir_emitter.EmitConstantGlobals()); in CompileAheadOfTime()
871 ir_emitter in CompileAheadOfTime()
880 ir_emitter.EmitComputation( in CompileAheadOfTime()
|
| D | BUILD | 89 ":ir_emitter",
258 name = "ir_emitter",
261 "ir_emitter.cc",
265 "ir_emitter.h",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 686 IrEmitterUnnested ir_emitter(module->config(), entry_computation, in RunBackend() local
689 TF_RETURN_IF_ERROR(ir_emitter.EmitConstantGlobals()); in RunBackend()
693 TF_RETURN_IF_ERROR(entry_computation->Accept(&ir_emitter)); in RunBackend()
767 ir_emitter.ConsumeThunkSequence(), std::move(stream_assignment), in RunBackend()
|
| D | BUILD | 146 name = "ir_emitter",
148 "ir_emitter.cc",
153 "ir_emitter.h",
826 ":ir_emitter",
|
| D | ir_emitter_unnested.cc | 2163 IrEmitterUnnested ir_emitter(hlo_module_config_, branch_computation, in BuildConditionalThunk() local
2165 TF_CHECK_OK(branch_computation->Accept(&ir_emitter)); in BuildConditionalThunk()
2166 branch_thunks.push_back(std::move(*ir_emitter.ConsumeThunkSequence())); in BuildConditionalThunk()
|