Home
last modified time | relevance | path

Searched refs:collection_name (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform_test.py451 collection_name = 'node_collection'
453 base_meta_graph_def.collection_def[collection_name].node_list.value.extend(
459 base_meta_graph_def, meta_graph_def, collection_name, removed_op_names)
461 collection = meta_graph_def.collection_def[collection_name]
467 collection_name = 'int_collection'
469 base_meta_graph_def.collection_def[collection_name].int64_list.value[:] = (
475 base_meta_graph_def, meta_graph_def, collection_name, removed_op_names)
477 collection = meta_graph_def.collection_def[collection_name]
484 collection_name = 'proto_collection'
486 base_meta_graph_def.collection_def[collection_name].any_list.value.extend([
[all …]
Dmeta_graph_transform.py464 collection_name, removed_op_names): argument
467 base_collection = base_meta_graph_def.collection_def[collection_name]
468 collection = meta_graph_def.collection_def[collection_name]
481 'In collection %s, nodes excluded are: %s', collection_name,
767 for collection_name in base_meta_graph_def.collection_def:
769 base_meta_graph_def, meta_graph_def, collection_name,
/external/python/apitools/apitools/base/py/testing/
Dmock.py246 def _MakeMockedService(api_name, collection_name, argument
257 _MockedMethod(api_name + '.' + collection_name + '.' + method,
312 collection_name = service_class._NAME
317 api_name, collection_name, self,
323 setattr(self, collection_name, mocked_service_class(self))
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py219 for collection_name in graph.get_all_collection_keys():
220 if collection_name not in WHITELIST_COLLECTIONS:
221 self._collections[collection_name] = graph.get_collection(
222 collection_name)
223 for collection_name in WHITELIST_COLLECTIONS:
224 self._collections[collection_name] = graph.get_collection_ref(
225 collection_name)
Dops.py6477 def register_proto_function(collection_name, argument
6502 collection_name)
6505 def get_collection_proto_type(collection_name): argument
6508 return _proto_function_registry.lookup(collection_name)[0]
6513 def get_to_proto_function(collection_name): argument
6516 return _proto_function_registry.lookup(collection_name)[1]
6521 def get_from_proto_function(collection_name): argument
6524 return _proto_function_registry.lookup(collection_name)[2]
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dexport.py44 def _get_first_op_from_collection(collection_name): argument
46 elements = ops.get_collection(collection_name)
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py141 for collection_name in [ops.GraphKeys.GLOBAL_VARIABLES,
143 mutable_collection = ops.get_collection_ref(collection_name)
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks.py294 collection_name = "{}_{}".format(_HOOKS, _STEPS_PER_RUN_VAR)
295 steps_per_run_vars = graph.get_collection(collection_name)
308 collections=[collection_name, ops.GraphKeys.LOCAL_VARIABLES],
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dgraph_actions.py413 def _get_first_op_from_collection(collection_name): argument
414 elements = ops.get_collection(collection_name)
/external/tensorflow/tensorflow/python/ops/
Dvariables.py2447 for collection_name in (ops.GraphKeys.GLOBAL_VARIABLES,
2449 for var in variable_op.graph.get_collection(collection_name):
/external/tensorflow/tensorflow/python/tpu/
Dtpu_estimator.py187 collection_name = '{}_{}'.format(_TPU_ESTIMATOR, _ITERATIONS_PER_LOOP_VAR)
188 iter_vars = graph.get_collection(collection_name)
203 collections=[collection_name, ops.GraphKeys.LOCAL_VARIABLES],
/external/libtextclassifier/annotator/
Dannotator.cc1387 {regex_pattern.config->collection_name()->str(), in RegexClassifyText()
2064 {regex_pattern.config->collection_name()->str(), in RegexChunk()
Dmodel.fbs214 collection_name:string;