Home
last modified time | relevance | path

Searched refs:add_to_collection (Results 1 – 25 of 55) 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.py433 ops.add_to_collection(ops.GraphKeys.SAVEABLE_OBJECTS, saveable)
449 ops.add_to_collection("iterator_ops", init_op)
454 ops.add_to_collection("iterator_ops", get_next.indices)
455 ops.add_to_collection("iterator_ops", get_next.values)
456 ops.add_to_collection("iterator_ops", get_next.dense_shape)
463 ops.add_to_collection("iterator_ops", get_next_list[i].indices)
464 ops.add_to_collection("iterator_ops", get_next_list[i].values)
465 ops.add_to_collection("iterator_ops", get_next_list[i].dense_shape)
467 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/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.py395 ops.add_to_collection(key, val)
638 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.py297 ops.add_to_collection(_SUMMARY_WRITER_INIT_COLLECTION_NAME, self._init_op)
730 … ops.add_to_collection(ops.GraphKeys._SUMMARY_COLLECTION, op) # pylint: disable=protected-access
782 … ops.add_to_collection(ops.GraphKeys._SUMMARY_COLLECTION, op) # pylint: disable=protected-access
814 … 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.py1833 ops_lib.add_to_collection("int_collection", 3)
1834 ops_lib.add_to_collection("float_collection", 3.5)
1835 ops_lib.add_to_collection("string_collection", "hello")
1836 ops_lib.add_to_collection("variable_collection", v0)
1841 ops_lib.add_to_collection("user_defined_string_collection",
1843 ops_lib.add_to_collection("user_defined_bytes_collection",
1847 ops_lib.add_to_collection("user_defined_any_collection", any_buf)
1880 ops_lib.add_to_collection(save, 3)
1886 ops_lib.add_to_collection("int_collection", 3)
1887 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.py868 graph.add_to_collection(key, variable)
873 graph.add_to_collection(
887 graph.add_to_collection(key, col_op)
893 graph.add_to_collection(key, int(value))
896 graph.add_to_collection(
1024 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/saved_model/
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.py119 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, initializer)
255 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, local_init_op)
/external/tensorflow/tensorflow/python/compiler/xla/
Djit.py140 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/compiler/mlir/tensorflow/tests/tf_saved_model/
Dhash_table_asset_v1.py62 tf.add_to_collection(tf.GraphKeys.ASSET_FILEPATHS, vocab_file_tensor)
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py281 g.add_to_collection(ops.GraphKeys.GLOBAL_STEP, clone.optimizer.iterations)
355 ops.add_to_collection(ops.GraphKeys.LOCAL_VARIABLES, v)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py745 dest_graph.add_to_collection(key, new_value)
757 dest_graph.add_to_collection(key, col_op)
763 dest_graph.add_to_collection(key, int(value))
766 dest_graph.add_to_collection(key,

123