| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | tf_mlir_translate.cc | 52 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToMlirImport() argument
66 specs.graph_as_function = graph_as_function; in GraphdefToMlirImport()
112 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToMlirTranslateFunction() argument
117 convert_legacy_fed_inputs, graph_as_function, upgrade_legacy, in GraphdefToMlirTranslateFunction()
130 bool convert_legacy_fed_inputs, bool graph_as_function, bool upgrade_legacy, in GraphdefToMlirTranslateFunction() argument
147 prune_unused_nodes, convert_legacy_fed_inputs, graph_as_function, in GraphdefToMlirTranslateFunction()
250 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToSplattedMlirTranslateFunction() argument
255 convert_legacy_fed_inputs, graph_as_function, upgrade_legacy, in GraphdefToSplattedMlirTranslateFunction()
299 bool convert_legacy_fed_inputs, bool graph_as_function, bool upgrade_legacy, in GraphdefToSplattedMlirTranslateFunction() argument
316 prune_unused_nodes, convert_legacy_fed_inputs, graph_as_function, in GraphdefToSplattedMlirTranslateFunction()
|
| H A D | tf_mlir_translate.h | 50 bool graph_as_function, bool upgrade_legacy,
65 bool convert_legacy_fed_inputs, bool graph_as_function, bool upgrade_legacy,
81 bool graph_as_function, bool upgrade_legacy,
97 bool convert_legacy_fed_inputs, bool graph_as_function, bool upgrade_legacy,
|
| H A D | tf_mlir_translate_registration.cc | 51 convert_legacy_fed_inputs, graph_as_function, upgrade_legacy, in GraphdefToMlirTranslateFunction()
65 convert_legacy_fed_inputs, graph_as_function, upgrade_legacy, in GraphdefToSplattedMlirTranslateFunction()
|
| H A D | tf_mlir_translate_cl.h | 41 extern llvm::cl::opt<bool> graph_as_function;
|
| H A D | mlir_roundtrip_flags.h | 71 bool graph_as_function = false; member
|
| H A D | tf_mlir_translate_cl.cc | 105 opt<bool> graph_as_function("tf-graph-as-function", variable
|
| H A D | mlir_roundtrip_flags.cc | 58 ss << "\ngraph_as_function: " << graph_as_function; in str()
|
| H A D | import_model.cc | 2432 if (specs.graph_as_function) { in Convert()
4247 specs.graph_as_function = true; in ConvertFunctionToMlir()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_graph_to_mlir_pass.cc | 74 import_config.graph_as_function = true; in RunOnGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| H A D | graphdef_to_tfl_flatbuffer.cc | 88 specs.graph_as_function = false; in ConvertGraphDefToTFLiteFlatBuffer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | mlir_tf2xla.cc | 123 specs.graph_as_function = false; in ConvertGraphDefToXlaViaMlir()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/ |
| H A D | mlir_graph_optimization_pass.cc | 214 import_config.graph_as_function = true; in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | compile_mlir_util.cc | 849 config.graph_as_function = true; in GraphToModule()
|