| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 341 Status LegalizeToHlo(mlir::ModuleOp module_op, llvm::StringRef device_type, in LegalizeToHlo() argument
344 mlir::PassManager tf2xla(module_op.getContext()); in LegalizeToHlo()
350 tensorflow::DumpMlirOpToFile("legalize_hlo_before", module_op); in LegalizeToHlo()
354 module_op.getContext()->disableMultithreading(); in LegalizeToHlo()
362 mlir::StatusScopedDiagnosticHandler error_handler(module_op.getContext()); in LegalizeToHlo()
364 if (failed(tf2xla.run(module_op))) { in LegalizeToHlo()
370 tensorflow::DumpMlirOpToFile("legalize_hlo_after", module_op); in LegalizeToHlo()
375 Status BuildHloFromTfInner(mlir::ModuleOp module_op, xla::XlaBuilder& builder, in BuildHloFromTfInner() argument
382 LegalizeToHlo(module_op, device_type, custom_legalization_passes)); in BuildHloFromTfInner()
384 mlir::Block& block = module_op.lookupSymbol<mlir::FuncOp>("main").front(); in BuildHloFromTfInner()
[all …]
|
| D | tf_xla_mlir_translate.cc | 241 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHloViaBuilder() argument
248 TF_RETURN_IF_ERROR(RefineShapes(arg_shapes, module_op)); in CompileMlirToXlaHloViaBuilder()
250 mlir::FuncOp main = module_op.lookupSymbol<mlir::FuncOp>("main"); in CompileMlirToXlaHloViaBuilder()
265 TF_RETURN_IF_ERROR(BuildHloFromTf(module_op, builder, xla_params, returns, in CompileMlirToXlaHloViaBuilder()
288 return PopulateResultIOInfo(module_op, arg_shapes, /*use_tuple_args=*/false, in CompileMlirToXlaHloViaBuilder()
294 mlir::ModuleOp module_op, llvm::raw_ostream& output, bool via_builder) { in MlirTfToHloTextTranslateFunctionImpl() argument
295 if (!module_op) return mlir::failure(); in MlirTfToHloTextTranslateFunctionImpl()
311 ? CompileMlirToXlaHloViaBuilder(module_op, arg_shapes, device_type, in MlirTfToHloTextTranslateFunctionImpl()
314 : CompileMlirToXlaHlo(module_op, arg_shapes, device_type, in MlirTfToHloTextTranslateFunctionImpl()
330 mlir::ModuleOp module_op, llvm::raw_ostream& output) { in MlirTfGraphToHloTextTranslateFunction() argument
[all …]
|
| D | compile_mlir_util.h | 71 mlir::ModuleOp module_op, llvm::StringRef device_type,
91 Status BuildHloFromTf(mlir::ModuleOp module_op, xla::XlaBuilder& builder,
103 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
112 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
136 mlir::ModuleOp module_op, llvm::ArrayRef<XlaArgument> args,
|
| D | serialize_mlir_module_utils.cc | 27 std::string SerializeMlirModule(mlir::ModuleOp module_op) { in SerializeMlirModule() argument
32 module_op.print(os, print_flags); in SerializeMlirModule()
|
| D | serialize_mlir_module_utils.h | 29 std::string SerializeMlirModule(mlir::ModuleOp module_op);
|
| /external/llvm-project/mlir/test/IR/ |
| D | invalid-module-op.mlir | 5 func @module_op() {
18 func @module_op() {
29 func @module_op() {
40 func @module_op() {
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_tf_ops_pass.cc | 178 void CreateFunctions(ModuleOp module_op, in CreateFunctions() argument
180 MLIRContext *context = module_op.getContext(); in CreateFunctions()
181 SymbolTable symbol_table(module_op); in CreateFunctions()
304 ModuleOp module_op = getOperation(); in runOnOperation() local
306 for (auto func_op : module_op.getOps<FuncOp>()) { in runOnOperation()
317 CreateFunctions(module_op, *metadatas); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 115 auto module_op = module.release(); in GetFromText() local
118 return absl::make_unique<TFRDecomposeContext>(module_op); in GetFromText()
|
| /external/llvm-project/mlir/lib/Bindings/Python/ |
| D | IRModules.cpp | 2761 PyOperationRef module_op = PyOperation::forOperation( in populateIRSubmodule() local
2764 PyBlock returnBlock(module_op, mlirModuleGetBody(self.get())); in populateIRSubmodule()
|