| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | bridge_logger_test.cc | 52 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
54 TF_ASSERT_OK(DeserializeMlirModule(module_with_add, &mlir_context, in TEST()
81 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
83 TF_ASSERT_OK(DeserializeMlirModule(module_with_add, &mlir_context, in TEST()
85 TF_ASSERT_OK(DeserializeMlirModule(module_with_sub, &mlir_context, in TEST()
121 mlir::MLIRContext mlir_context(mlir_registry); in TEST() local
123 TF_ASSERT_OK(DeserializeMlirModule(module_with_add, &mlir_context, in TEST()
|
| D | serialize_mlir_module_utils.cc | 37 mlir::MLIRContext* mlir_context, in DeserializeMlirModule() argument
45 mlir::StatusScopedDiagnosticHandler error_handler(mlir_context); in DeserializeMlirModule()
49 mlir_context); in DeserializeMlirModule()
|
| D | serialize_mlir_module_utils.h | 35 mlir::MLIRContext* mlir_context,
|
| D | compile_mlir_util.h | 172 mlir::MLIRContext& mlir_context, llvm::ArrayRef<xla::XlaOp> xla_params,
|
| D | compile_mlir_util.cc | 674 mlir::MLIRContext mlir_context(mlir_registry); in CompileSerializedMlirToXlaHlo() local
678 DeserializeMlirModule(mlir_module_string, &mlir_context, &mlir_module)); in CompileSerializedMlirToXlaHlo()
862 mlir::MLIRContext& mlir_context, llvm::ArrayRef<xla::XlaOp> xla_params, in BuildHloFromGraph() argument
870 GraphToModule(graph, control_rets, flib_def, debug_info, &mlir_context)); in BuildHloFromGraph()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 44 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext() argument
51 mlir_context_(mlir_context), in IrEmitterContext()
70 mlir::MLIRContext* mlir_context() { return mlir_context_; } in mlir_context() function
|
| D | gpu_compiler.cc | 1007 mlir::MLIRContext mlir_context(registry); in CompileModuleToLlvmIrImpl() local
1009 mlir::ModuleOp::create(mlir::Builder(&mlir_context).getUnknownLoc()); in CompileModuleToLlvmIrImpl()
1028 mlir::PassManager pm(&mlir_context); in CompileModuleToLlvmIrImpl()
1038 &mlir_context, results->llvm_module.get()); in CompileModuleToLlvmIrImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 896 mlir::MLIRContext& mlir_context) { in LowerMLIRModule() argument
897 LoadMLIRDialects(mlir_context); in LowerMLIRModule()
898 mlir::PassManager pm(&mlir_context); in LowerMLIRModule()
1000 HloModule* module, mlir::MLIRContext& mlir_context, in createMLIRModule() argument
1002 LoadMLIRDialects(mlir_context); in createMLIRModule()
1003 mlir::OpBuilder builder(&mlir_context); in createMLIRModule()
1026 mlir::IntegerType::get(&mlir_context, 64), in createMLIRModule()
1125 mlir::MLIRContext mlir_context; in CompileLegacyCpuExecutable() local
1126 LoadMLIRDialects(mlir_context); in CompileLegacyCpuExecutable()
1192 createMLIRModule(module.get(), mlir_context, assignment.get())); in CompileLegacyCpuExecutable()
[all …]
|
| D | dot_op_emitter.cc | 130 llvm::IRBuilder<>* b, mlir::MLIRContext* mlir_context,
251 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in DotOpEmitter() argument
261 mlir_context_(mlir_context), in DotOpEmitter()
1241 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitNonBatchDotOperation() argument
1250 executable_run_options_value, b, mlir_context, in EmitNonBatchDotOperation()
1326 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in PotentiallyImplementedAsEigenMatmul() argument
1382 mlir::MLIRContext* mlir_context, const HloModuleConfig& hlo_module_config, in EmitBatchDotOperation() argument
1392 b, mlir_context, hlo_module_config, target_machine_features, in EmitBatchDotOperation()
1396 executable_run_options_value, b, mlir_context, in EmitBatchDotOperation()
1457 executable_run_options_value, b, mlir_context, hlo_module_config, in EmitBatchDotOperation()
[all …]
|
| D | dot_op_emitter.h | 66 llvm::IRBuilder<>* b, mlir::MLIRContext* mlir_context,
|
| D | ir_emitter.h | 87 IrEmitter(mlir::MLIRContext* mlir_context, const HloModule& hlo_module,
|
| D | ir_emitter.cc | 95 IrEmitter::IrEmitter(mlir::MLIRContext* mlir_context, in IrEmitter() argument
111 mlir_context_(mlir_context), in IrEmitter()
|
| /external/tensorflow/tensorflow/tools/tfg_graph_transforms/ |
| D | tfg_graph_transforms_main.cc | 132 mlir::MLIRContext* mlir_context) { in ImportModel() argument
141 return mlir::tfg::ImportSavedModelToMlir(mlir_context, debug_info, in ImportModel()
149 return mlir::tfg::ImportGraphDef(mlir_context, debug_info, graph_def); in ImportModel()
|
| /external/tensorflow/tensorflow/core/transforms/const_dedupe_hoist/ |
| D | pass.cc | 50 mlir_context = context; in initialize()
75 MLIRContext* mlir_context; member
175 name_id, StringAttr::get(mlir_context, "const_dedupe_hoist/" + in BuildIdentity()
|