experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Execute TODO to explain graph-consumer usage of

Browse Source 
RemoveNewDefaultAttrsFromGraphDef().
Change: 115506523
This commit is contained in:
Josh Levenberg 2016-02-24 16:22:53 -08:00 committed by TensorFlower Gardener
parent 8041c546bb
commit 2cc5ed87e3
1 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tensorflow/core/framework/graph_def_util.h
View File

@ -65,7 +65,8 @@ Status AddDefaultAttrsToGraphDef(GraphDef* graph_def,
// compatible. If not nulllptr, the op/attr pairs that were removed
// are added to '*op_attr_removed'.
//
// Expected usage:
// Expected usage, for a producer that wants to prepare a graph for
// a consumer:
// // For each consumer, update 'graph_def':
// OpListOpRegistry consumer_op_registry(consumer_server_op_list);
// std::unordered_set<std::pair<string, string>> op_attr_removed;
@ -78,8 +79,12 @@ Status AddDefaultAttrsToGraphDef(GraphDef* graph_def,
// // Consumer can use 'graph_def', and 'op_attr_removed' summarizes
// // what changes had to be made to 'graph_def' for it to work.
//
// TODO(josh11b): Describe how to use this function on the consumer using the
// stripped_op_list field from a producer.
// Expected usage, for a consumer that has a graph and a
// (optionally-stripped) op_list from a producer (say from a call to
// StrippedOpListForGraph(), or in the MetaGraphDef):
// OpListOpRegistry producer_op_registry(producer_stripped_op_list);
// TF_RETURN_IF_ERROR(RemoveNewDefaultAttrsFromGraphDef(
// &graph_def, *OpRegistry::Global(), producer_op_registry, nullptr));
Status RemoveNewDefaultAttrsFromGraphDef(
GraphDef* graph_def, const OpRegistryInterface& consumer_op_registry,
const OpRegistryInterface& producer_op_registry,
@ -87,13 +92,14 @@ Status RemoveNewDefaultAttrsFromGraphDef(
// Collect the ops used by a graph.
//
// This function computes the stripped_op_list field of MetaGraphDef and similar
// protos. The op_registry should contain the ops used to produce graph_def,
// and stripped_op_list can be used as the producer_op_registry argument to
// RemoveNewDefaultAttrsFromGraphDef to improve forwards compatibility
// (using OpListOpRegistry to turn the OpList into an OpRegistryInterface).
// This function computes the stripped_op_list field of MetaGraphDef
// and similar protos. The op_registry should contain the ops used to
// produce graph_def. The resulting stripped_op_list can be
// communicated from the producer to the consumer, which can use
// RemoveNewDefaultAttrsFromGraphDef() to improve forwards compatibility
// (using an OpListOpRegistry as indicated in the example above).
//
// Most users will pass OpRegistry::Global() for op_registry to strip against
// Most users will pass *OpRegistry::Global() for op_registry to strip against
// the list of ops registered in this process.
Status StrippedOpListForGraph(const GraphDef& graph_def,
const OpRegistryInterface& op_registry,