Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/ops/losses/
Dutil_test.py37 ops.add_to_collection(
39 ops.add_to_collection(
46 ops.add_to_collection(
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Ddataset_serialization_test_base.py434 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
450 ops.add_to_collection("iterator_ops", init_op)
455 ops.add_to_collection("iterator_ops", get_next.indices)
456 ops.add_to_collection("iterator_ops", get_next.values)
457 ops.add_to_collection("iterator_ops", get_next.dense_shape)
464 ops.add_to_collection("iterator_ops", get_next_list[i].indices)
465 ops.add_to_collection("iterator_ops", get_next_list[i].values)
466 ops.add_to_collection("iterator_ops", get_next_list[i].dense_shape)
468 ops.add_to_collection("iterator_ops", get_next_list[i])
Dcheckpoint_input_pipeline_hook_test.py56 ops.add_to_collection('my_vars', global_step)
57 ops.add_to_collection('my_vars', latest_feature)
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer_test.py68 ops.add_to_collection('a2', a2) # Explicitly add to collection.
75 ops.add_to_collection('train_op', d) # d is the fetch node.
119 ops.add_to_collection('train_op', buf_shape)
120 ops.add_to_collection('train_op', f_shape)
/external/tensorflow/tensorflow/python/eager/
Dfunction_defun_collection_test.py46 ops.add_to_collection('x', x)
47 ops.add_to_collection('y', y)
54 ops.add_to_collection('z', 7)
/external/tensorflow/tensorflow/python/saved_model/
Dsaved_model_test.py79 ops.add_to_collection(ops.GraphKeys.ASSET_FILEPATHS, asset_file_tensor)
488 ops.add_to_collection("foo_vars", v)
498 ops.add_to_collection("bar_vars", v)
845 ops.add_to_collection("v", v1)
847 ops.add_to_collection("v", v2)
851 ops.add_to_collection("v", v3)
881 ops.add_to_collection("v", v1)
883 ops.add_to_collection("v", v2)
909 ops.add_to_collection("v", v1)
911 ops.add_to_collection("v", v2)
[all …]
Dbuilder_impl.py492 ops.add_to_collection(constants.MAIN_OP_KEY, main_op)
511 ops.add_to_collection(constants.TRAIN_OP_KEY, train_op)
792 ops.add_to_collection(constants.ASSETS_KEY, asset_any_proto)
Dload_v1_in_v2.py116 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, initializer)
238 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, local_init_op)
/external/tensorflow/tensorflow/python/summary/
Dplugin_asset.py79 graph.add_to_collection(name, instance)
80 graph.add_to_collection(_PLUGIN_ASSET_PREFIX, plugin_asset_cls.plugin_name)
/external/tensorflow/tensorflow/python/ops/
Dresources.py55 ops.add_to_collection(ops.GraphKeys.RESOURCES, resource)
57 ops.add_to_collection(ops.GraphKeys.LOCAL_RESOURCES, resource)
Dsummary_op_util.py39 ops.add_to_collection(key, val)
Dlogging_ops.py390 ops.add_to_collection(key, val)
633 ops.add_to_collection(ops.GraphKeys.SUMMARY_OP, summary_op)
Dtensor_forest_ops.py48 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, self)
Dsummary_ops_v2.py244 ops.add_to_collection(_SUMMARY_WRITER_INIT_COLLECTION_NAME, self._init_op)
677 … ops.add_to_collection(ops.GraphKeys._SUMMARY_COLLECTION, op) # pylint: disable=protected-access
729 … ops.add_to_collection(ops.GraphKeys._SUMMARY_COLLECTION, op) # pylint: disable=protected-access
761 … ops.add_to_collection(ops.GraphKeys._SUMMARY_COLLECTION, op) # pylint: disable=protected-access
Dboosted_trees_ops.py126 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, self._saveable)
226 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, self._saveable)
/external/tensorflow/tensorflow/python/training/
Dsupervisor.py409 ops.add_to_collection(ops.GraphKeys.READY_OP, ready_op)
431 ops.add_to_collection(ops.GraphKeys.INIT_OP, init_op)
454 ops.add_to_collection(ops.GraphKeys.LOCAL_INIT_OP, local_init_op)
468 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
483 ops.add_to_collection(ops.GraphKeys.SUMMARY_OP, summary_op)
499 ops.add_to_collection(ops.GraphKeys.GLOBAL_STEP, global_step)
Dsaver_test.py1834 ops_lib.add_to_collection("int_collection", 3)
1835 ops_lib.add_to_collection("float_collection", 3.5)
1836 ops_lib.add_to_collection("string_collection", "hello")
1837 ops_lib.add_to_collection("variable_collection", v0)
1842 ops_lib.add_to_collection("user_defined_string_collection",
1844 ops_lib.add_to_collection("user_defined_bytes_collection",
1848 ops_lib.add_to_collection("user_defined_any_collection", any_buf)
1881 ops_lib.add_to_collection(save, 3)
1887 ops_lib.add_to_collection("int_collection", 3)
1888 ops_lib.add_to_collection("int_collection", 3.5)
[all …]
Dsaver_test_utils.py45 ops_lib.add_to_collection(ops_lib.GraphKeys.SAVEABLE_OBJECTS,
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph.py863 graph.add_to_collection(key, variable)
868 graph.add_to_collection(
882 graph.add_to_collection(key, col_op)
888 graph.add_to_collection(key, int(value))
891 graph.add_to_collection(
1019 graph.add_to_collection(unbound_inputs_col_name, k)
Dframework_lib.py43 from tensorflow.python.framework.ops import add_to_collection
/external/tensorflow/tensorflow/python/compiler/xla/
Djit.py112 ops.add_to_collection(_XLA_SCOPE_KEY, xla_scope_counter)
/external/tensorflow/tensorflow/core/protobuf/
Dmeta_graph.proto101 // tf.add_to_collection("your_collection_name", your_simple_value)
105 // 1) tf.add_to_collection("your_collection_name",
119 // 2) tf.add_to_collection("your_collection_name", str(your_proto))
133 // tf.add_to_collection("your_collection_name",
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py48 graph.add_to_collection('train_op', fetch)
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py280 g.add_to_collection(ops.GraphKeys.GLOBAL_STEP, clone.optimizer.iterations)
354 ops.add_to_collection(ops.GraphKeys.LOCAL_VARIABLES, v)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py764 dest_graph.add_to_collection(key, new_value)
776 dest_graph.add_to_collection(key, col_op)
782 dest_graph.add_to_collection(key, int(value))
785 dest_graph.add_to_collection(key,

123