Searched defs:mlir_context (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 45 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext()
69 mlir::MLIRContext* mlir_context() { return mlir_context_; } in mlir_context() function
|
| D | gpu_compiler.cc | 591 mlir::MLIRContext mlir_context; in CompileModuleToLlvmIrImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | serialize_mlir_module_utils.cc | 37 mlir::MLIRContext* mlir_context, in DeserializeMlirModule()
|
| D | compile_mlir_util.cc | 499 mlir::MLIRContext mlir_context(mlir_registry); in CompileSerializedMlirToXlaHlo() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 240 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in DotOpEmitter()
1044 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation()
1128 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation()
1236 llvm::IRBuilder<>* b, mlir::MLIRContext* mlir_context, in EmitDotOperation()
|
| D | cpu_compiler.cc | 675 mlir::MLIRContext mlir_context; in RunBackend() local
893 mlir::MLIRContext mlir_context; in CompileAheadOfTime() local
|
| D | ir_emitter.cc | 92 mlir::MLIRContext* mlir_context, const HloModule& hlo_module, in IrEmitter()
|