Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/ops/losses/
Dutil_test.py33 ops.add_to_collection(
35 ops.add_to_collection(
42 ops.add_to_collection(
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test_base.py516 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
532 ops.add_to_collection("iterator_ops", init_op)
537 ops.add_to_collection("iterator_ops", get_next.indices)
538 ops.add_to_collection("iterator_ops", get_next.values)
539 ops.add_to_collection("iterator_ops", get_next.dense_shape)
546 ops.add_to_collection("iterator_ops", get_next_list[i].indices)
547 ops.add_to_collection("iterator_ops", get_next_list[i].values)
548 ops.add_to_collection("iterator_ops", get_next_list[i].dense_shape)
550 ops.add_to_collection("iterator_ops", get_next_list[i])
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer_test.py65 ops.add_to_collection('a2', a2) # Explicitly add to collection.
72 ops.add_to_collection('train_op', d) # d is the fetch node.
116 ops.add_to_collection('train_op', buf_shape)
117 ops.add_to_collection('train_op', f_shape)
/external/tensorflow/tensorflow/python/eager/
Dfunction_defun_collection_test.py42 ops.add_to_collection('x', x)
43 ops.add_to_collection('y', y)
50 ops.add_to_collection('z', 7)
/external/tensorflow/tensorflow/python/summary/
Dplugin_asset.py75 graph.add_to_collection(name, instance)
76 graph.add_to_collection(_PLUGIN_ASSET_PREFIX, plugin_asset_cls.plugin_name)
/external/tensorflow/tensorflow/python/ops/
Dresources.py51 ops.add_to_collection(ops.GraphKeys.RESOURCES, resource)
53 ops.add_to_collection(ops.GraphKeys.LOCAL_RESOURCES, resource)
Dsummary_ops_v2.py316 ops.add_to_collection(_SUMMARY_WRITER_INIT_COLLECTION_NAME, self._init_op)
414 ops.add_to_collection(_SUMMARY_WRITER_INIT_COLLECTION_NAME, init_op)
772 … ops.add_to_collection(ops.GraphKeys._SUMMARY_COLLECTION, op) # pylint: disable=protected-access
825 … ops.add_to_collection(ops.GraphKeys._SUMMARY_COLLECTION, op) # pylint: disable=protected-access
857 … ops.add_to_collection(ops.GraphKeys._SUMMARY_COLLECTION, op) # pylint: disable=protected-access
Dsummary_op_util.py35 ops.add_to_collection(key, val)
Dlogging_ops.py407 ops.add_to_collection(key, val)
650 ops.add_to_collection(ops.GraphKeys.SUMMARY_OP, summary_op)
Dboosted_trees_ops.py123 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, self._saveable)
223 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, self._saveable)
/external/tensorflow/tensorflow/python/training/
Dsupervisor.py405 ops.add_to_collection(ops.GraphKeys.READY_OP, ready_op)
427 ops.add_to_collection(ops.GraphKeys.INIT_OP, init_op)
450 ops.add_to_collection(ops.GraphKeys.LOCAL_INIT_OP, local_init_op)
464 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
479 ops.add_to_collection(ops.GraphKeys.SUMMARY_OP, summary_op)
495 ops.add_to_collection(ops.GraphKeys.GLOBAL_STEP, global_step)
Dsaver_test.py1899 ops_lib.add_to_collection("int_collection", 3)
1900 ops_lib.add_to_collection("float_collection", 3.5)
1901 ops_lib.add_to_collection("string_collection", "hello")
1902 ops_lib.add_to_collection("variable_collection", v0)
1907 ops_lib.add_to_collection("user_defined_string_collection",
1909 ops_lib.add_to_collection("user_defined_bytes_collection",
1913 ops_lib.add_to_collection("user_defined_any_collection", any_buf)
1946 ops_lib.add_to_collection(save, 3)
1952 ops_lib.add_to_collection("int_collection", 3)
1953 ops_lib.add_to_collection("int_collection", 3.5)
[all …]
Dsaver_test_utils.py41 ops_lib.add_to_collection(ops_lib.GraphKeys.SAVEABLE_OBJECTS,
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph.py871 graph.add_to_collection(key, variable)
876 graph.add_to_collection(
890 graph.add_to_collection(key, col_op)
896 graph.add_to_collection(key, int(value))
899 graph.add_to_collection(
1029 graph.add_to_collection(unbound_inputs_col_name, k)
Dframework_lib.py39 from tensorflow.python.framework.ops import add_to_collection
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcheckpoint_input_pipeline_hook_test.py48 ops.add_to_collection('my_vars', global_step)
49 ops.add_to_collection('my_vars', latest_feature)
/external/tensorflow/tensorflow/python/compiler/xla/
Djit.py136 ops.add_to_collection(_XLA_SCOPE_KEY, xla_scope_counter)
/external/tensorflow/tensorflow/python/saved_model/
Dbuilder_impl.py495 ops.add_to_collection(constants.MAIN_OP_KEY, main_op)
514 ops.add_to_collection(constants.TRAIN_OP_KEY, train_op)
797 ops.add_to_collection(constants.ASSETS_KEY, asset_any_proto)
Dload_v1_in_v2.py120 ops.add_to_collection("saved_model_initializers", initializer)
258 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, local_init_op)
/external/tensorflow/tensorflow/core/protobuf/
Dmeta_graph.proto98 // tf.add_to_collection("your_collection_name", your_simple_value)
102 // 1) tf.add_to_collection("your_collection_name",
116 // 2) tf.add_to_collection("your_collection_name", str(your_proto))
130 // tf.add_to_collection("your_collection_name",
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py44 graph.add_to_collection('train_op', fetch)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dhash_table_asset_v1.py58 tf.add_to_collection(tf.GraphKeys.ASSET_FILEPATHS, vocab_file_tensor)
/external/tensorflow/tensorflow/python/trackable/
Dasset.py100 ops.add_to_collection(ops.GraphKeys.ASSET_FILEPATHS, asset.asset_path)
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py279 g.add_to_collection(ops.GraphKeys.GLOBAL_STEP, clone.optimizer.iterations)
353 ops.add_to_collection(ops.GraphKeys.LOCAL_VARIABLES, v)
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dlookup_ops.py98 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, init_op)

123