本文整理匯總了Python中tensorflow.python.framework.ops.get_from_proto_function方法的典型用法代碼示例。如果您正苦於以下問題：Python ops.get_from_proto_function方法的具體用法？Python ops.get_from_proto_function怎麽用？Python ops.get_from_proto_function使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類tensorflow.python.framework.ops的用法示例。
在下文中一共展示了ops.get_from_proto_function方法的1個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: _handle_collection_def
# 需要導入模塊: from tensorflow.python.framework import ops [as 別名]
# 或者: from tensorflow.python.framework.ops import get_from_proto_function [as 別名]
def _handle_collection_def(multi_gpu_meta_graph_def, op_names_to_replicate,
num_replicas):
allow_bytes_list_keys = [tf.GraphKeys.QUEUE_RUNNERS,
tf.GraphKeys.GLOBAL_VARIABLES,
tf.GraphKeys.TRAINABLE_VARIABLES,
tf.GraphKeys.MOVING_AVERAGE_VARIABLES,
tf.GraphKeys.LOCAL_VARIABLES,
tf.GraphKeys.MODEL_VARIABLES,
tf.GraphKeys.GRADIENTS_INFO,
tf.GraphKeys.GLOBAL_STEP]
keys_to_remove = []
for key, col_def in multi_gpu_meta_graph_def.collection_def.items():
kind = col_def.WhichOneof("kind")
# Update node_list collections (e.g., GLOBAL_STEP, TRAIN_OP, UPDATE_OP,
# LOSSES, ...)
if kind == 'node_list':
new_col_def = get_new_col_def_of_node_list(
col_def, op_names_to_replicate, num_replicas)
multi_gpu_meta_graph_def.collection_def[key].Clear()
multi_gpu_meta_graph_def.collection_def[key].CopyFrom(new_col_def)
elif kind == 'bytes_list':
if ops.get_from_proto_function(key):
# Collections in allow_bytes_list_keys will be handled
# explicitly below
# (e.g., QUEUE_RUNNERS, LOCAL_VARIABLES, ...)
if key in allow_bytes_list_keys:
continue
# Remove unhandled collections (e.g., COND_CONTEXT)
# TODO: Handle all protos in tf.GraphKeys
else:
keys_to_remove.append(key)
# Keep collections without proto function
# (e.g., user defined string)
else:
continue
else:
raise RuntimeError("Should not reach here")
for key in keys_to_remove:
del multi_gpu_meta_graph_def.collection_def[key]
# Update QUEUE_RUNNERS and LOCAL_VARIABLES collection
update_queue_runners(multi_gpu_meta_graph_def, op_names_to_replicate,
num_replicas)
update_local_variables(multi_gpu_meta_graph_def, op_names_to_replicate,
num_replicas)
update_shard_info_for_in_graph(multi_gpu_meta_graph_def, num_replicas)