Searched refs:graph_as_function (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| 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()
110 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToMlirTranslateFunction() argument
115 convert_legacy_fed_inputs, graph_as_function, upgrade_legacy, in GraphdefToMlirTranslateFunction()
128 bool convert_legacy_fed_inputs, bool graph_as_function, bool upgrade_legacy, in GraphdefToMlirTranslateFunction() argument
144 prune_unused_nodes, convert_legacy_fed_inputs, graph_as_function, in GraphdefToMlirTranslateFunction()
224 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference, in GraphdefToSplattedMlirTranslateFunction() argument
229 convert_legacy_fed_inputs, graph_as_function, upgrade_legacy, in GraphdefToSplattedMlirTranslateFunction()
272 bool convert_legacy_fed_inputs, bool graph_as_function, bool upgrade_legacy, in GraphdefToSplattedMlirTranslateFunction() argument
288 prune_unused_nodes, convert_legacy_fed_inputs, graph_as_function, in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate.h | 48 bool graph_as_function, bool upgrade_legacy,
62 bool convert_legacy_fed_inputs, bool graph_as_function, bool upgrade_legacy,
76 bool graph_as_function, bool upgrade_legacy, bool enable_shape_inference,
89 bool convert_legacy_fed_inputs, bool graph_as_function, bool upgrade_legacy,
|
| D | tf_mlir_translate_registration.cc | 49 convert_legacy_fed_inputs, graph_as_function, upgrade_legacy, in GraphdefToMlirTranslateFunction()
63 convert_legacy_fed_inputs, graph_as_function, upgrade_legacy, in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate_cl.h | 40 extern llvm::cl::opt<bool> graph_as_function;
|
| D | mlir_roundtrip_flags.h | 56 bool graph_as_function = false; member
|
| D | tf_mlir_translate_cl.cc | 104 opt<bool> graph_as_function("tf-graph-as-function", variable
|
| D | import_model.cc | 2226 if (specs.graph_as_function) { in Convert()
3908 specs.graph_as_function = true; in ConvertFunctionToMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | dump_graph.cc | 79 import_config.graph_as_function = true; in DumpTextualIRToFile()
|
| D | compile_mlir_util.cc | 661 config.graph_as_function = true; in GraphToModule()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| D | graphdef_to_tfl_flatbuffer.cc | 78 specs.graph_as_function = false; in ConvertGraphDefToTFLiteFlatBuffer()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_tf2xla.cc | 123 specs.graph_as_function = false; in ConvertGraphDefToXlaViaMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 193 import_config.graph_as_function = true; in Run()
|