Searched defs:mlir_context (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | bridge_logger_test.cc | 52 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
81 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
121 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
|
| D | serialize_mlir_module_utils.cc | 37 mlir::MLIRContext* mlir_context, in DeserializeMlirModule()
|
| D | compile_mlir_util.cc | 525 mlir::MLIRContext mlir_context(mlir_registry); in CompileSerializedMlirToXlaHlo() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 44 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext()
68 mlir::MLIRContext* mlir_context() { return mlir_context_; } in mlir_context() function
|
| D | gpu_compiler.cc | 781 mlir::MLIRContext mlir_context; in CompileModuleToLlvmIrImpl() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 238 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in DotOpEmitter()
1063 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation()
1147 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation()
1255 llvm::IRBuilder<>* b, mlir::MLIRContext* mlir_context, in EmitDotOperation()
|
| D | cpu_compiler.cc | 699 mlir::MLIRContext mlir_context; in RunBackend() local
927 mlir::MLIRContext mlir_context; in CompileAheadOfTime() local
|
| D | ir_emitter.cc | 92 mlir::MLIRContext* mlir_context, const HloModule& hlo_module, in IrEmitter()
|