Home
last modified time | relevance | path

Searched refs:TRAINABLE_VARIABLES (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/python/profiler/
Dtfprof_logger.py35 TRAINABLE_VARIABLES = '_trainable_variables' variable
129 for v in graph.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES):
133 entry.types.append(TRAINABLE_VARIABLES)
136 logged_ops[v.op.name].types.append(TRAINABLE_VARIABLES)
Doption_builder.py104 'account_type_regexes': [tfprof_logger.TRAINABLE_VARIABLES],
/external/tensorflow/tensorflow/contrib/copy_graph/python/util/
Dcopy_elements.py75 name == ops.GraphKeys.TRAINABLE_VARIABLES or scope == ''):
83 ops.GraphKeys.TRAINABLE_VARIABLES))
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_ops_benchmark.py137 ops.GraphKeys.TRAINABLE_VARIABLES)
161 ops.GraphKeys.TRAINABLE_VARIABLES)
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py274 trainable_vars = graph1.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
286 trainable_vars = graph2.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
616 [ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.TRAINABLE_VARIABLES]
648 ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.TRAINABLE_VARIABLES
653 omit_collection_keys=[ops.GraphKeys.TRAINABLE_VARIABLES])
655 include_collection_keys=[ops.GraphKeys.TRAINABLE_VARIABLES],
660 ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.TRAINABLE_VARIABLES
/external/tensorflow/tensorflow/python/ops/
Dtemplate.py310 ops.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES))
318 ops.GraphKeys.TRAINABLE_VARIABLES)
418 return ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES,
Dresource_variable_ops.py514 if trainable and ops.GraphKeys.TRAINABLE_VARIABLES not in collections:
515 collections = list(collections) + [ops.GraphKeys.TRAINABLE_VARIABLES]
1540 ops.GraphKeys.TRAINABLE_VARIABLES,
Dvariables.py1529 if trainable and ops.GraphKeys.TRAINABLE_VARIABLES not in collections:
1530 collections = list(collections) + [ops.GraphKeys.TRAINABLE_VARIABLES]
2801 return ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES, scope)
/external/tensorflow/tensorflow/contrib/tfprof/
Dmodel_analyzer.py49 'account_type_regexes': [tfprof_logger.TRAINABLE_VARIABLES],
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_tpu_variables.py288 collections.append(ops.GraphKeys.TRAINABLE_VARIABLES)
289 l = g.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES)
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py125 ops.GraphKeys.TRAINABLE_VARIABLES)
141 ops.GraphKeys.TRAINABLE_VARIABLES)
292 self.assertLen(graph.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES), 1)
Dwrap_function.py73 if v.trainable and ops.GraphKeys.TRAINABLE_VARIABLES not in collections:
74 collections = list(collections) + [ops.GraphKeys.TRAINABLE_VARIABLES]
/external/tensorflow/tensorflow/contrib/layers/
DREADME.md28 `tf.GraphKeys.TRAINABLE_VARIABLES`.
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-graph-keys.pbtxt114 name: "TRAINABLE_VARIABLES"
/external/tensorflow/tensorflow/python/layers/
Dcore_test.py83 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
112 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 1)
127 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 0)
238 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
Dbase_test.py102 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES))
116 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 1)
/external/tensorflow/tensorflow/python/keras/layers/
Dkernelized_test.py127 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES),
263 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES),
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy.py338 collections.append(ops.GraphKeys.TRAINABLE_VARIABLES)
339 l = g.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES)
Dtpu_strategy.py116 var_collections.append(ops.GraphKeys.TRAINABLE_VARIABLES)
117 l = g.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES)
Dmirrored_strategy.py271 collections.append(ops.GraphKeys.TRAINABLE_VARIABLES)
272 l = g.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES)
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py733 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
752 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
866 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
934 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
1000 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
Dvariable_scope_test.py258 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES),
268 self.assertFalse(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES))
1121 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES))
1169 for v in scope1.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1184 for v in scope2.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1205 for v in scope.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
/external/tensorflow/tensorflow/python/training/
Dwarm_starting_util.py333 ops.GraphKeys.TRAINABLE_VARIABLES,
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py1754 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1765 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1776 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1786 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2389 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2400 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2411 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2421 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
5246 ops.GraphKeys.TRAINABLE_VARIABLES)
5327 for v in ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
[all …]
Dfeature_column_v2_test.py1942 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1954 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1965 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1976 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2757 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2768 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2779 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2789 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
6767 for v in ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
6860 trainable_vars = ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
[all …]

12