Searched refs:ir_emitter_context (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 40 IrEmitterContext* ir_emitter_context) in IrEmitterNested() argument
41 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/true), in IrEmitterNested()
47 IrEmitterContext* ir_emitter_context) { in Create() argument
49 hlo_module_config, nested_computation, ir_emitter_context)); in Create()
|
| D | ir_emitter_nested.h | 45 IrEmitterContext* ir_emitter_context);
69 IrEmitterContext* ir_emitter_context);
|
| D | ir_emitter.cc | 80 IrEmitterContext* ir_emitter_context, bool is_nested) in IrEmitter() argument
81 : ir_emitter_context_(ir_emitter_context), in IrEmitter()
82 module_(ir_emitter_context->llvm_module()), in IrEmitter()
84 bindings_(ir_emitter_context->hlo_module(), in IrEmitter()
85 &ir_emitter_context->buffer_assignment(), &b_, module_, in IrEmitter()
|
| D | gpu_compiler.cc | 596 IrEmitterContext ir_emitter_context( in CompileModuleToLlvmIrImpl() local
606 &ir_emitter_context)); in CompileModuleToLlvmIrImpl()
658 *constants = std::move(ir_emitter_context.constants()); in CompileModuleToLlvmIrImpl()
1102 IrEmitterContext* ir_emitter_context) { in CompileLmhloToExecutable() argument
1115 ir_emitter_context->set_allocations(allocations); in CompileLmhloToExecutable()
1120 ir_emitter_context)); in CompileLmhloToExecutable()
1147 std::move(ir_emitter_context->constants()), std::move(output_info), in CompileLmhloToExecutable()
|
| D | gpu_compiler.h | 173 IrEmitterContext* ir_emitter_context);
|
| D | ir_emitter.h | 117 IrEmitterContext* ir_emitter_context, bool is_nested);
|
| D | ir_emitter_unnested.h | 152 IrEmitterContext* ir_emitter_context);
242 IrEmitterContext* ir_emitter_context);
|
| D | ir_emitter_unnested.cc | 533 IrEmitterContext* ir_emitter_context) in IrEmitterUnnested() argument
534 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/false) {} in IrEmitterUnnested()
539 IrEmitterContext* ir_emitter_context) { in Create() argument
541 hlo_module_config, hlo_computation, ir_emitter_context)); in Create()
544 mlir::Builder(ir_emitter_context->mlir_context()).getUnknownLoc())); in Create()
|
| D | BUILD | 243 "ir_emitter_context.h",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 67 IrEmitterContext ir_emitter_context( in CompileMlirModule() local
78 std::move(llvm_module), &ir_emitter_context); in CompileMlirModule()
|