Home
last modified time | relevance | path

Searched refs:prune_unused_nodes (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc51 bool prune_unused_nodes, bool convert_legacy_fed_inputs, in GraphdefToMlirImport() argument
64 specs.prune_unused_nodes = prune_unused_nodes; in GraphdefToMlirImport()
77 if (specs.prune_unused_nodes) { in GraphdefToMlirImport()
98 specs.prune_unused_nodes ? pruned_graph_def : graphdef, debug_info, specs, in GraphdefToMlirImport()
109 bool prune_unused_nodes, bool convert_legacy_fed_inputs, in GraphdefToMlirTranslateFunction() argument
114 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
127 absl::string_view control_output_arrays, bool prune_unused_nodes, in GraphdefToMlirTranslateFunction() argument
144 prune_unused_nodes, convert_legacy_fed_inputs, graph_as_function, in GraphdefToMlirTranslateFunction()
223 bool prune_unused_nodes, bool convert_legacy_fed_inputs, in GraphdefToSplattedMlirTranslateFunction() argument
228 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
[all …]
Dtf_mlir_translate.h47 bool prune_unused_nodes, bool convert_legacy_fed_inputs,
61 absl::string_view control_output_arrays, bool prune_unused_nodes,
75 bool prune_unused_nodes, bool convert_legacy_fed_inputs,
88 absl::string_view control_output_arrays, bool prune_unused_nodes,
Dtf_mlir_translate_registration.cc48 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToMlirTranslateFunction()
62 output_arrays, control_output_arrays, prune_unused_nodes, in GraphdefToSplattedMlirTranslateFunction()
Dtf_mlir_translate_cl.h38 extern llvm::cl::opt<bool> prune_unused_nodes;
Dmlir_roundtrip_flags.h49 bool prune_unused_nodes = false; member
Dtf_mlir_translate_cl.cc91 opt<bool> prune_unused_nodes( variable
Dimport_model.cc850 if (specs_.prune_unused_nodes) { in AddNodesToShapeRefiner()
1409 if (specs_.prune_unused_nodes) { in PrepareConvert()
2227 if (specs.prune_unused_nodes || !specs.inputs.empty() || in Convert()
3272 specs.prune_unused_nodes = true; in Convert()
3627 specs.prune_unused_nodes = true; in ConvertGraph()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc77 absl::string_view output_arrays, bool prune_unused_nodes, in LoadFromGraphdefOrMlirSource() argument
97 prune_unused_nodes, /*convert_legacy_fed_inputs=*/true, in LoadFromGraphdefOrMlirSource()
Dtf_to_tfjs_json.h43 absl::string_view output_arrays, bool prune_unused_nodes,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddump_graph.cc80 import_config.prune_unused_nodes = false; in DumpTextualIRToFile()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc125 specs.prune_unused_nodes, /*convert_legacy_fed_inputs=*/true, in LoadFromGraphdefOrMlirSource()
132 specs.prune_unused_nodes, /*convert_legacy_fed_inputs=*/true, in LoadFromGraphdefOrMlirSource()
Dtf_tfl_translate.cc150 specs.prune_unused_nodes = true; in main()
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc76 specs.prune_unused_nodes = true; in ConvertGraphDefToTFLiteFlatBuffer()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_tf2xla.cc121 specs.prune_unused_nodes = false; in ConvertGraphDefToXlaViaMlir()