| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | serialize_mlir_module_utils.cc | 37 mlir::MLIRContext* mlir_context, in DeserializeMlirModule() argument
45 mlir::StatusScopedDiagnosticHandler error_handler(mlir_context); in DeserializeMlirModule()
48 *mlir_module = mlir::parseSourceString(serialized_mlir_module, mlir_context); in DeserializeMlirModule()
|
| D | serialize_mlir_module_utils.h | 34 mlir::MLIRContext* mlir_context,
|
| D | compile_mlir_util.cc | 499 mlir::MLIRContext mlir_context(mlir_registry); in CompileSerializedMlirToXlaHlo() local
503 DeserializeMlirModule(mlir_module_string, &mlir_context, &mlir_module)); in CompileSerializedMlirToXlaHlo()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 45 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext() argument
52 mlir_context_(mlir_context), 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
592 mlir_context.loadDialect<mlir::lmhlo::LmhloDialect, mlir::mhlo::MhloDialect, in CompileModuleToLlvmIrImpl()
598 cuda_compute_capability, profile_index_map, &mlir_context, in CompileModuleToLlvmIrImpl()
|
| D | ir_emitter_unnested.cc | 544 mlir::Builder(ir_emitter_context->mlir_context()).getUnknownLoc())); in Create()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 130 llvm::IRBuilder<>* b, mlir::MLIRContext* mlir_context,
240 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in DotOpEmitter() argument
250 mlir_context_(mlir_context), in DotOpEmitter()
1044 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation() argument
1053 executable_run_options_value, b, mlir_context, in EmitNonBatchDotOperation()
1128 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation() argument
1186 executable_run_options_value, b, mlir_context, hlo_module_config, in EmitBatchDotOperation()
1236 llvm::IRBuilder<>* b, mlir::MLIRContext* mlir_context, in EmitDotOperation() argument
1246 executable_run_options_value, b, mlir_context, in EmitDotOperation()
1252 executable_run_options_value, b, mlir_context, in EmitDotOperation()
|
| D | dot_op_emitter.h | 67 llvm::IRBuilder<>* b, mlir::MLIRContext* mlir_context,
|
| D | cpu_compiler.cc | 675 mlir::MLIRContext mlir_context; in RunBackend() local
676 LoadMLIRDialects(mlir_context); in RunBackend()
737 IrEmitter ir_emitter(&mlir_context, *module, *assignment, llvm_module.get(), in RunBackend()
893 mlir::MLIRContext mlir_context; in CompileAheadOfTime() local
894 LoadMLIRDialects(mlir_context); in CompileAheadOfTime()
945 IrEmitter ir_emitter(&mlir_context, *module, *assignment, &llvm_module, in CompileAheadOfTime()
|
| D | ir_emitter.h | 84 IrEmitter(mlir::MLIRContext* mlir_context, const HloModule& hlo_module,
|
| D | ir_emitter.cc | 92 mlir::MLIRContext* mlir_context, const HloModule& hlo_module, in IrEmitter() argument
102 mlir_context_(mlir_context), in IrEmitter()
|