| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | dump_mlir_util_test.cc | 32 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
42 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
52 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
76 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
88 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
100 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
|
| D | device_util_test.cc | 67 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
106 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
115 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
124 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
135 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
146 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
160 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
181 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
|
| D | tpu_rewrite_device_util_test.cc | 631 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
649 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
668 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
687 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
704 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
722 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
742 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
762 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
785 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
811 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in TEST() local
[all …]
|
| D | tf_xla_mlir_translate.cc | 396 mlir::OwningOpRef<mlir::ModuleOp> module_ref; in SerializedMlirStringAttrToMlirModuleTranslate() local
|
| /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
238 mlir::OwningOpRef<mlir::ModuleOp> module_ref = std::move(module).value(); in QuantizePtqModelPreCalibration() local
310 mlir::OwningOpRef<mlir::ModuleOp> module_ref = std::move(module).value(); in QuantizePtqModelPostCalibration() local
393 mlir::OwningOpRef<mlir::ModuleOp> module_ref = std::move(module).value(); in QuantizePtqDynamicRange() local
|
| /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
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 246 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in Run() local
389 mlir::OwningOpRef<mlir::ModuleOp> module_ref = in Run() local
|
| /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
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | tf_to_tfrt_data.cc | 287 mlir::OwningOpRef<mlir::ModuleOp> *module_ref) { in TFDataGraphDefToTFDataMLIR()
339 mlir::OwningOpRef<mlir::ModuleOp> module_ref; in TFDataGraphDefToHostBEF() local
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | insert_quantized_functions.cc | 144 OwningOpRef<ModuleOp> module_ref = in runOnOperation() local
|
| /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
|