| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | device_util_test.cc | 67 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
81 AddDevicesToOp(*module_ref, &device_set); in TEST()
84 (*module_ref)->getAttrOfType<mlir::DictionaryAttr>("tf.devices"); in TEST()
106 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
109 AddDevicesToOp(*module_ref, /*device_set=*/nullptr); in TEST()
110 EXPECT_EQ((*module_ref)->getAttr("tf.devices"), nullptr); in TEST()
115 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
119 EXPECT_TRUE(mlir::succeeded(GetDevicesFromOp(*module_ref, &devices))); in TEST()
124 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
126 mlir::Builder builder(*module_ref); in TEST()
[all …]
|
| D | dump_mlir_util_test.cc | 32 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
36 std::string filepath = DumpMlirOpToFile("module", module_ref.get()); in TEST()
42 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
46 std::string filepath = DumpMlirOpToFile("module", module_ref.get()); in TEST()
52 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
58 module_ref->getOperation()->print(os, in TEST()
63 std::string filepath = DumpMlirOpToFile("module", module_ref.get()); in TEST()
76 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
82 DumpCrashReproducerToFile("module", pm, module_ref.get()); in TEST()
88 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
[all …]
|
| D | tpu_rewrite_device_util_test.cc | 631 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
633 mlir::OpBuilder builder(module_ref->getBodyRegion()); in TEST()
649 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
651 mlir::OpBuilder builder(module_ref->getBodyRegion()); in TEST()
668 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
670 mlir::OpBuilder builder(module_ref->getBodyRegion()); in TEST()
687 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
689 mlir::OpBuilder builder(module_ref->getBodyRegion()); in TEST()
704 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
706 mlir::OpBuilder builder(module_ref->getBodyRegion()); in TEST()
[all …]
|
| D | tf_xla_mlir_translate.cc | 396 mlir::OwningOpRef<mlir::ModuleOp> module_ref; in SerializedMlirStringAttrToMlirModuleTranslate() local
398 DeserializeMlirModule(str_attr.getValue().str(), context, &module_ref); in SerializedMlirStringAttrToMlirModuleTranslate()
404 return module_ref; in SerializedMlirStringAttrToMlirModuleTranslate()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 246 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in Run() local
248 AddDevicesToOp(*module_ref, &device_set); in Run()
263 DumpModule(*module_ref, llvm::formatv("mlir_{0}_before_", name)); in Run()
271 pass_status = pass_registration.pass->Run(config_proto, *module_ref, in Run()
279 auto module_ref_clone = module_ref->clone(); in Run()
286 module_ref = module_ref_clone; in Run()
314 DumpModule(*module_ref, llvm::formatv("mlir_{0}_after_", name)); in Run()
324 Status status = ConvertMlirToGraph(*module_ref, export_config, graph, in Run()
389 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in Run() local
391 AddDevicesToOp(*module_ref, options.device_set); in Run()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/ |
| D | quantize_model.cc | 151 mlir::OwningOpRef<mlir::ModuleOp> module_ref = std::move(module).value(); in QuantizeQatModel() local
154 module_ref.get(), &context, bundle ? bundle->GetSession() : nullptr); in QuantizeQatModel()
198 if (failed(pm.run(*module_ref))) { in QuantizeQatModel()
204 return ConvertMlirModuleToGraphDef(*module_ref); in QuantizeQatModel()
238 mlir::OwningOpRef<mlir::ModuleOp> module_ref = std::move(module).value(); in QuantizePtqModelPreCalibration() local
241 module_ref.get(), &context, bundle ? bundle->GetSession() : nullptr); in QuantizePtqModelPreCalibration()
261 if (failed(pm.run(*module_ref))) { in QuantizePtqModelPreCalibration()
267 return ConvertMlirModuleToGraphDef(*module_ref); in QuantizePtqModelPreCalibration()
310 mlir::OwningOpRef<mlir::ModuleOp> module_ref = std::move(module).value(); in QuantizePtqModelPostCalibration() local
344 if (failed(pm.run(*module_ref))) { in QuantizePtqModelPostCalibration()
[all …]
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_graph_to_mlir_pass.cc | 88 module_ref = ConvertGraphToMlir(*input_graph, debug_info, *flib_def, in RunOnGraph() local
90 if (!module_ref.ok()) in RunOnGraph()
94 module_ref.status().error_message()) in RunOnGraph()
97 mlir::ModuleOp module = module_ref.ValueOrDie().get(); in RunOnGraph()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | insert_quantized_functions.cc | 144 OwningOpRef<ModuleOp> module_ref = in runOnOperation() local
154 if (failed(pm.run(*module_ref))) { in runOnOperation()
163 for (func::FuncOp func : module_ref->getOps<func::FuncOp>()) { in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | tf_to_tfrt_data.cc | 287 mlir::OwningOpRef<mlir::ModuleOp> *module_ref) { in TFDataGraphDefToTFDataMLIR() argument
300 TF_ASSIGN_OR_RETURN(*module_ref, ConvertGraphdefToMlir( in TFDataGraphDefToTFDataMLIR()
339 mlir::OwningOpRef<mlir::ModuleOp> module_ref; in TFDataGraphDefToHostBEF() local
341 TFDataGraphDefToTFDataMLIR(graph_def, &mlir_ctx, &module_ref)); in TFDataGraphDefToHostBEF()
342 TF_RETURN_IF_ERROR(CompileTFDataMLIRToBEF(module_ref.get(), bef)); in TFDataGraphDefToHostBEF()
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/saved_model/ |
| D | saved_model_roundtrip_test.cc | 49 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in RunRoundTrip() local
53 auto status = mlir::tfg::ExportMlirToSavedModel(*module_ref, original_model, in RunRoundTrip()
|
| /external/tensorflow/tensorflow/tools/tfg_graph_transforms/ |
| D | tfg_graph_transforms_main.cc | 217 auto module_ref = std::move(module_ref_status.ValueOrDie()); in main() local
222 RunOptimizationPasses(pass_pipeline, *module_ref, &context); in main()
229 ExportTFGModule(*module_ref, data_format, input_file, output_file); in main()
|
| /external/python/cpython3/Lib/test/ |
| D | test_struct.py | 696 module_ref = weakref.ref(_struct_module)
703 module_ref(), "_struct module was not garbage collected")
|