Home
last modified time | relevance | path

Searched refs:collection_def (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dgrappler_item_builder.cc312 meta_graph.collection_def().count("train_op") > 0) { in GrapplerItemFromMetaGraphDef()
313 const CollectionDef& nodes = meta_graph.collection_def().at("train_op"); in GrapplerItemFromMetaGraphDef()
443 if (meta_graph.collection_def().count(var_collection) == 0) { in GrapplerItemFromMetaGraphDef()
446 const CollectionDef& vars = meta_graph.collection_def().at(var_collection); in GrapplerItemFromMetaGraphDef()
456 if (meta_graph.collection_def().count("table_initializer") > 0) { in GrapplerItemFromMetaGraphDef()
458 meta_graph.collection_def().at("table_initializer"); in GrapplerItemFromMetaGraphDef()
480 if (meta_graph.collection_def().count("saved_model_assets") > 0) { in GrapplerItemFromMetaGraphDef()
482 meta_graph.collection_def().at("saved_model_assets"); in GrapplerItemFromMetaGraphDef()
509 } else if (meta_graph.collection_def().count("asset_filepaths") > 0) { in GrapplerItemFromMetaGraphDef()
511 meta_graph.collection_def().at("asset_filepaths"); in GrapplerItemFromMetaGraphDef()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dloader_impl.py135 collection_def = meta_graph_def_to_load.collection_def
142 elif constants.ASSETS_KEY in collection_def:
143 assets_any_proto = collection_def[constants.ASSETS_KEY].any_list.value
182 collection_def = meta_graph_def_to_load.collection_def
184 if init_op_key in collection_def:
185 init_op_list = collection_def[init_op_key].node_list.value
H A Dsave.py816 meta_graph_def.collection_def[constants.MAIN_OP_KEY].node_list.value.append(
H A Dsaved_model_test.py1422 self._validate_asset_collection(export_dir, foo_graph.collection_def,
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dtrt_convert.py562 collection_def = self._grappler_meta_graph_def.collection_def["train_op"]
563 denylist = collection_def.node_list.value
612 input_meta_graph_def.collection_def):
627 input_meta_graph_def.collection_def):
628 self._grappler_meta_graph_def.collection_def[collection_key].CopyFrom(
629 input_meta_graph_def.collection_def[collection_key])
786 collection_def = src_meta_graph_def.collection_def[key]
787 kind = collection_def.WhichOneof("kind")
796 for value in collection_def.bytes_list.value:
805 field = getattr(collection_def, kind)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/inputs/
H A Dfile_input_yielder.cc86 if (metagraph.collection_def().count("train_op") == 0 || in NextItem()
87 !metagraph.collection_def().at("train_op").has_node_list() || in NextItem()
88 metagraph.collection_def().at("train_op").node_list().value_size() == 0) { in NextItem()
96 metagraph.collection_def().at("train_op").node_list().value()) { in NextItem()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dmeta_graph.proto80 // collection_def: Map from collection name to collections.
82 map<string, CollectionDef> collection_def = 4; field
105 // collection_def {
118 // collection_def {
133 // collection_def {
159 // collection_def {
174 // collection_def {
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dmeta_graph.py411 col_def = meta_graph_def.collection_def[key]
442 if key in meta_graph_def.collection_def:
443 del meta_graph_def.collection_def[key]
770 for key, col_def in meta_graph_def.collection_def.items():
822 if ops.GraphKeys.TRAINABLE_VARIABLES in meta_graph_def.collection_def:
825 meta_graph_def.collection_def[ops.GraphKeys.TRAINABLE_VARIABLES]))
826 for key, value in sorted(meta_graph_def.collection_def.items()):
H A Dmeta_graph_test.py321 orig_meta_graph.collection_def[ops.GraphKeys.METRIC_VARIABLES].CopyFrom(
322 orig_meta_graph.collection_def["variables"])
518 del a.collection_def["unbound_inputs"]
519 del b.collection_def["unbound_inputs"]
H A Dconvert_to_constants.py1031 for raw in meta_graph.collection_def["variables"].bytes_list.value:
1036 meta_graph.collection_def[name].bytes_list.value[:] = raw_list
1042 meta_graph.collection_def["train_op"].CopyFrom(fetch_collection)
H A Dtest_util.py270 tester.assertEqual(set(a.collection_def), set(b.collection_def))
271 collection_keys = a.collection_def.keys()
273 a_value = a.collection_def[k]
274 b_value = b.collection_def[k]
H A Dconvert_to_constants_test.py1035 meta_graph.collection_def["train_op"].CopyFrom(fetch_collection)
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dconvert_saved_model.py192 collection_def = meta_graph.collection_def
193 if constants.ASSETS_KEY in collection_def:
H A Dutil.py257 meta_graph.collection_def["train_op"].CopyFrom(fetch_collection)
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/
H A Dloader_util.cc48 const auto& collection_def_map = meta_graph_def.collection_def(); in GetInitOp()
79 const auto& collection_def_map = meta_graph_def.collection_def(); in GetAssetFileDefs()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dsaver_test.py1923 collection_def = meta_graph_def.collection_def
1924 self.assertEqual(len(collection_def), 12)
1948 self.assertEqual(len(meta_graph_def.collection_def), 0)
1955 self.assertEqual(len(meta_graph_def.collection_def), 0)
1986 collection_def = meta_graph_def.collection_def["savers"]
1987 kind = collection_def.WhichOneof("kind")
1990 savers = getattr(collection_def, kind)
1994 collection_def = meta_graph_def0.collection_def["savers"]
1995 kind = collection_def.WhichOneof("kind")
1998 savers = getattr(collection_def, kind)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Drewrite_utils.cc282 CollectionDef collection_def; in GetGrapplerItem() local
283 auto node_list = collection_def.mutable_node_list(); in GetGrapplerItem()
285 (*meta_graph_def.mutable_collection_def())["train_op"] = collection_def; in GetGrapplerItem()
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/
H A Dsaved_model.pbtxt2461 collection_def {
2469 collection_def {
2477 collection_def {
2488 collection_def {
2498 collection_def {
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/graph_analyzer/
H A Dgraph_analyzer_tool.cc50 metagraph.collection_def().at("train_op").node_list().value()) { in MaybePruneGraph()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-meta-graph-def.pbtxt27 name: "collection_def"
/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dcost_analyzer_tool.py73 meta_graph.collection_def["train_op"].CopyFrom(fetch_collection)
H A Ditem_wrapper.cc137 if (metagraph.collection_def().count("train_op") == 0) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/testdata/
H A Dmetrics_export_meta_graph.pb1547 collection_def {
1556 collection_def {
/aosp_15_r20/external/tensorflow/tensorflow/python/tools/
H A Dsaved_model_aot_compile.py424 new_meta_graph_def.collection_def['train_op'].CopyFrom(fetch_collection)
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dmeta_graph.pb.h612 collection_def() const; in Swap()
3406 MetaGraphDef::collection_def() const { in collection_def() function