Searched refs:base_meta_graph_def (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
D | meta_graph_transform_test.py | 59 base_meta_graph_def = saver.export_meta_graph() 81 print('AAAAAA: {}'.format(base_meta_graph_def)) 83 base_meta_graph_def, input_names, output_names, transforms, tags) 374 base_meta_graph_def = meta_graph_pb2.MetaGraphDef() 380 base_meta_graph_def.graph_def.node.extend([node_def]) 386 base_meta_graph_def.graph_def.node.extend([node_def]) 388 base_meta_graph_def.collection_def[ 391 base_meta_graph_def.collection_def[ 397 expected_meta_graph_def.CopyFrom(base_meta_graph_def) 408 base_meta_graph_def.graph_def.node.extend([node_def]) [all …]
|
D | meta_graph_transform.py | 387 def _find_all_mandatory_retain_ops(base_meta_graph_def): argument 410 if base_meta_graph_def.HasField('saver_def'): 411 saver_defs.append(base_meta_graph_def.saver_def) 413 base_meta_graph_def, _ops.GraphKeys.SAVERS)) 430 variables = _get_all_protos_from_collection(base_meta_graph_def, var_coll) 439 base_meta_graph_def, _ops.GraphKeys.TABLE_INITIALIZERS) 455 base_meta_graph_def, op_coll) 463 def _add_pruned_collection(base_meta_graph_def, meta_graph_def, argument 467 base_collection = base_meta_graph_def.collection_def[collection_name] 494 def _add_pruned_saver(base_meta_graph_def, meta_graph_def, removed_op_names): argument [all …]
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
D | estimator.py | 1444 base_meta_graph_def = builder._saved_model.meta_graphs[0] 1468 base_meta_graph_def, input_names, output_names,
|