Home
last modified time | relevance | path

Searched defs:meta_graph_def (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/
H A Dreader_test.cc46 void CheckMetaGraphDef(const MetaGraphDef& meta_graph_def) { in CheckMetaGraphDef()
58 MetaGraphDef meta_graph_def; in TEST_F() local
67 MetaGraphDef meta_graph_def; in TEST_F() local
80 MetaGraphDef meta_graph_def; in TEST_F() local
93 MetaGraphDef meta_graph_def; in TEST_F() local
102 MetaGraphDef meta_graph_def; in TEST_F() local
117 MetaGraphDef meta_graph_def; in TEST_F() local
131 MetaGraphDef meta_graph_def; in TEST_F() local
H A Dloader_util.cc31 Status GetInitOp(const string& export_dir, const MetaGraphDef& meta_graph_def, in GetInitOp()
68 Status GetAssetFileDefs(const MetaGraphDef& meta_graph_def, in GetAssetFileDefs()
H A Dreader.cc88 MetaGraphDef* meta_graph_def) { in FindMetaGraphDef()
119 MetaGraphDef* const meta_graph_def) { in ReadMetaGraphDefFromSavedModel()
H A Dbundle_v2.h47 MetaGraphDef& meta_graph_def() { return meta_graph_def_; } in meta_graph_def() function
H A Dloader.h66 MetaGraphDef meta_graph_def; member
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dloader_impl.py193 def _get_op_from_collection(meta_graph_def, op_key): argument
197 def _get_op_from_signature_def(meta_graph_def, op_signature_key, import_scope): argument
207 def get_init_op(meta_graph_def, import_scope=None): argument
214 def get_train_op(meta_graph_def, import_scope=None): argument
H A Dbuilder_impl.py115 def _save_and_write_assets(self, meta_graph_def, assets_list=None): argument
134 def _tag_and_add_meta_graph(self, meta_graph_def, tags, signature_def_map): argument
748 def _add_asset_to_metagraph(meta_graph_def, asset_filename, asset_tensor): argument
H A Dsave.py771 def _fill_meta_graph_def(meta_graph_def, saveable_view, signature_functions, argument
1368 def _build_meta_graph_impl(obj, signatures, options, meta_graph_def=None): argument
1413 def _build_meta_graph(obj, signatures, options, meta_graph_def=None): argument
H A Dload_v1_in_v2.py91 def load_graph(self, returns, meta_graph_def): argument
137 def _extract_signatures(self, wrapped, meta_graph_def): argument
H A Dutils_impl.py286 def swap_function_tensor_content(meta_graph_def, from_endiness, to_endiness): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model.cc266 const tensorflow::MetaGraphDef& meta_graph_def) { in FindNamesForValidSignatures()
304 mlir::MLIRContext* context, const tensorflow::MetaGraphDef& meta_graph_def, in ImportSavedModel()
439 tensorflow::MetaGraphDef meta_graph_def; in ReadSavedModel() local
483 auto meta_graph_def = ReadSavedModel(saved_model_dir, tags); in LoadSavedModel() local
602 Options options, tensorflow::MetaGraphDef meta_graph_def, in SavedModelImpl()
H A Dsaved_model_import_input.cc26 const FallbackState& fallback_state, const MetaGraphDef* meta_graph_def, in Create()
45 const MetaGraphDef* meta_graph_def, const GraphDebugInfo& debug_info, in TfrtSavedModelMLIRImportInput()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dimport_model.h105 SavedModelMLIRImportInput(const MetaGraphDef* meta_graph_def, in SavedModelMLIRImportInput()
114 const MetaGraphDef& meta_graph_def() const { return *meta_graph_def_; } in meta_graph_def() function
/aosp_15_r20/external/tensorflow/tensorflow/python/tools/
H A Dsaved_model_cli.py95 def _get_inputs_tensor_info_from_meta_graph_def(meta_graph_def, argument
120 def _get_outputs_tensor_info_from_meta_graph_def(meta_graph_def, argument
357 def scan_meta_graph_def(meta_graph_def): argument
H A Dsaved_model_aot_compile.py311 def aot_compile_cpu_meta_graph_def(checkpoint_path, argument
413 def _optimize_graph(meta_graph_def, signature_def): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.cc64 Status RunGrappler(const MetaGraphDef& meta_graph_def, in RunGrappler()
155 Status RunTfTrt(const MetaGraphDef& meta_graph_def, in RunTfTrt()
429 Status InlineFunctions(const MetaGraphDef& meta_graph_def, in InlineFunctions()
H A Dtrt_convert_api_test.cc250 MetaGraphDef meta_graph_def = GetModel(); in ConvertAndRunFrozenGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dmeta_graph.py372 def add_collection_def(meta_graph_def, key, graph=None, argument
461 def strip_graph_default_valued_attrs(meta_graph_def): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dsavedmodel_export.cc41 tensorflow::MetaGraphDef meta_graph_def = original_saved_model.meta_graphs(0); in ExportMlirToSavedModel() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/tools/
H A Dfreeze_saved_model_test.cc55 MetaGraphDef* meta_graph_def = &saved_model_bundle->meta_graph_def; in AddSignatureDefToSavedModelBundle() local
80 MetaGraphDef* meta_graph_def = &saved_model_bundle->meta_graph_def; in AddGraphDefToSavedModelBundle() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dbyte_swap.cc148 Status ByteSwapTensorContent(MetaGraphDef* meta_graph_def) { in ByteSwapTensorContent()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dsaver.py1458 def _add_collection_def(meta_graph_def, key, export_scope=None): argument
1613 def _create_saver_from_imported_meta_graph(meta_graph_def, import_scope, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/summary/writer/
H A Dfake_summary_writer.py116 def add_meta_graph(self, meta_graph_def, global_step=None): argument
H A Dwriter.py225 def add_meta_graph(self, meta_graph_def, global_step=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Devent.proto41 bytes meta_graph_def = 9; field

12