Home
last modified time | relevance | path

Searched refs:variable_scope (Results 1 – 25 of 194) sorted by relevance

12345678

/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dvariable_scope_shim_test.py41 from tensorflow.python.ops import variable_scope
78 vs = variable_scope._get_default_variable_store()
86 vs = variable_scope._get_default_variable_store()
97 vs = variable_scope._get_default_variable_store()
110 with variable_scope.variable_scope("tower0") as tower:
111 with variable_scope.variable_scope("foo", initializer=init):
112 v = variable_scope.get_variable("v", [])
115 with variable_scope.variable_scope(tower, initializer=init):
116 w = variable_scope.get_variable("w", [])
124 with variable_scope.variable_scope("tower1") as tower:
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/variables/
Dvariable_scope_test.py37 from tensorflow.python.ops import variable_scope
79 vs = variable_scope._get_default_variable_store()
87 vs = variable_scope._get_default_variable_store()
94 vs = variable_scope._get_default_variable_store()
112 vs = variable_scope._get_default_variable_store()
125 with variable_scope.variable_scope("tower0") as tower:
126 with variable_scope.variable_scope("foo", initializer=init):
127 v = variable_scope.get_variable("v", [])
130 with variable_scope.variable_scope(tower, initializer=init):
131 w = variable_scope.get_variable("w", [])
[all …]
Dpartitioned_variables_test.py29 from tensorflow.python.ops import variable_scope
40 with variable_scope.variable_scope("root", partitioner=partitioner):
41 v0 = variable_scope.get_variable(
51 with variable_scope.variable_scope("root", partitioner=partitioner):
52 v0 = variable_scope.get_variable("v0", dtype=dtypes.int64, shape=[20])
59 with variable_scope.variable_scope(
61 v0 = variable_scope.get_variable(
78 with variable_scope.variable_scope("root", partitioner=partitioner):
79 v0 = variable_scope.get_variable(
175 with variable_scope.variable_scope(
[all …]
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils_test.py33 from tensorflow.python.ops import variable_scope
44 v1 = variable_scope.get_variable("var1", [1, 10])
45 v2 = variable_scope.get_variable("var2", [10, 10])
46 v3 = variable_scope.get_variable("var3", [100, 100])
47 with variable_scope.variable_scope("useful_scope"):
48 v4 = variable_scope.get_variable("var4", [9, 9])
63 with variable_scope.variable_scope("scope"):
64 v1 = variable_scope.get_variable(
144 with variable_scope.variable_scope("some_scope"):
145 my1 = variable_scope.get_variable("my1", [1, 10])
[all …]
Dwarm_starting_util_test.py27 from tensorflow.python.ops import variable_scope
61 var = variable_scope.get_variable(
81 all_vars.append(variable_scope.get_variable(
99 with variable_scope.variable_scope("", partitioner=partitioner):
120 fruit_weights = variable_scope.get_variable(
138 fruit_weights = variable_scope.get_variable(
152 fruit_weights = variable_scope.get_variable(
178 fruit_weights = variable_scope.get_variable(
207 fruit_weights = variable_scope.get_variable(
227 fruit_output_layer = variable_scope.get_variable(
[all …]
Dslot_creator.py43 from tensorflow.python.ops import variable_scope
59 current_partitioner = variable_scope.get_variable_scope().partitioner
60 variable_scope.get_variable_scope().set_partitioner(None)
70 slot = variable_scope.get_variable(
78 variable_scope.get_variable_scope().set_partitioner(current_partitioner)
149 with variable_scope.variable_scope(None, prefix + "/" + name):
208 with variable_scope.variable_scope(None, prefix + "/" + name):
Dmoving_averages.py25 from tensorflow.python.ops import variable_scope
151 with variable_scope.variable_scope(name, "WeightedMovingAvg",
153 value_x_weight_var = variable_scope.get_variable(
160 weight_var = variable_scope.get_variable(
231 with variable_scope.variable_scope(
239 if variable_scope.get_variable_scope().reuse:
243 for x in variable_scope.get_variable_scope().global_variables()
245 full_name = variable_scope.get_variable_scope().name + "/" + name
254 biased_var = variable_scope.get_variable(
260 local_step = variable_scope.get_variable(
Dcheckpoint_ops_test.py27 from tensorflow.python.ops import variable_scope
47 with variable_scope.variable_scope('some_scope'):
48 variable_scope.get_variable(name='embeddings', shape=[5, 16],
148 remapped_matrix = variable_scope.get_variable(
181 remapped_matrix = variable_scope.get_variable(
218 remapped_matrix = variable_scope.get_variable(
253 remapped_matrix = variable_scope.get_variable(
286 remapped_embeddings = variable_scope.get_variable(
332 remapped_embeddings = variable_scope.get_variable(
370 remapped_embeddings = variable_scope.get_variable(
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_variable_test.py39 from tensorflow.python.ops import variable_scope
106 v = variable_scope.variable(2.0, name="bar")
111 v1 = variable_scope.variable(1.0, name="foo")
143 v = variable_scope.variable(1.0, name="foo")
154 v = variable_scope.variable(1.0)
167 vs.append(variable_scope.variable(1.0, name="foo" + str(i)))
180 vs.append(variable_scope.variable(1.0, name="foo/bar"))
181 vs.append(variable_scope.variable(1.0, name="foo_1/bar"))
182 vs.append(variable_scope.variable(1.0, name="foo_1/bar_1"))
183 vs.append(variable_scope.variable(1.0, name="foo/bar_1"))
[all …]
Dshared_variable_creator_test.py20 from tensorflow.python.ops import variable_scope
55 with variable_scope.variable_creator_scope(creator_fns[0]):
56 v0 = variable_scope.variable(1.0, name="foo")
58 with variable_scope.variable_creator_scope(creator_fns[1]):
59 v1 = variable_scope.variable(1.0, name="foo")
61 with variable_scope.variable_creator_scope(creator_fns[2]):
62 v2 = variable_scope.variable(1.0, name="foo")
Dcheckpoint_utils_test.py31 from tensorflow.python.ops import variable_scope
41 v1 = variable_scope.get_variable("var1", [1, 10])
42 v2 = variable_scope.get_variable("var2", [10, 10])
79 v1 = variable_scope.get_variable("new_var1", [1, 10])
80 v2 = variable_scope.get_variable(
82 synchronization=variable_scope.VariableSynchronization.ON_READ,
83 aggregation=variable_scope.VariableAggregation.MEAN)
115 v1 = variable_scope.get_variable("new_var1", [1, 10])
Dmirrored_strategy_test.py56 from tensorflow.python.ops import variable_scope
368 with variable_scope.variable_creator_scope(thread_creator_fn):
370 v = variable_scope.variable(1.0)
383 variable_scope.variable_creator_scope(main_thread_creator):
576 c = variable_scope.variable(1.0, name="c")
580 b = variable_scope.variable(1.0, name="b")
587 a = variable_scope.variable(1.0, name="a")
605 c = variable_scope.get_variable("c", [1])
609 b = variable_scope.get_variable("b", [1])
616 a = variable_scope.get_variable("a", [1])
[all …]
Ddistribute_lib_test.py32 from tensorflow.python.ops import variable_scope
166 "bar", variable_scope.VariableSynchronization.AUTO,
167 variable_scope.VariableAggregation.NONE)
169 variable_scope.variable(1.0, name="bar"))
186 "baz", variable_scope.VariableSynchronization.AUTO,
187 variable_scope.VariableAggregation.NONE)
189 variable_scope.variable(1.0, name="baz"))
216 with variable_scope.variable_creator_scope(creator):
233 with variable_scope.variable_scope("AA"):
244 "baz", variable_scope.VariableSynchronization.ON_WRITE,
[all …]
Dparameter_server_strategy_test.py52 from tensorflow.python.ops import variable_scope
128 n = variable_scope.get_variable('n', initializer=10.0)
147 x = variable_scope.get_variable(
149 aggregation=variable_scope.VariableAggregation.SUM)
161 y = variable_scope.get_variable(
163 aggregation=variable_scope.VariableAggregation.SUM)
171 z = variable_scope.get_variable(
173 aggregation=variable_scope.VariableAggregation.SUM)
193 u = variable_scope.get_variable('u', initializer=30.0)
194 v = variable_scope.get_variable('v', initializer=30.0)
[all …]
Dreduce_util.py19 from tensorflow.python.ops import variable_scope
39 variable_scope.VariableAggregation.SUM: ReduceOp.SUM,
40 variable_scope.VariableAggregation.MEAN: ReduceOp.MEAN,
Dvars_test.py41 from tensorflow.python.ops import variable_scope
113 v = variable_scope.variable(
127 v_to_assign = variable_scope.variable(
129 v_to_assign_sub = variable_scope.variable(
158 v = variable_scope.variable(
211 v = variable_scope.variable(
270 v = variable_scope.variable(
294 v = variable_scope.variable(
346 v = variable_scope.variable(
402 v = variable_scope.variable(
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dtemplate_test.py28 from tensorflow.python.ops import variable_scope
36 return variable_scope.get_variable(
42 with variable_scope.variable_scope(scope_name):
43 return variable_scope.get_variable(
50 return variable_scope.get_variable(
56 variable_scope.get_variable(name, shape=[1], trainable=trainable)
57 return variable_scope.get_variable(
62 variable_scope.get_local_variable(
64 return variable_scope.get_variable(
83 m = variable_scope.get_variable(
[all …]
/external/tensorflow/tensorflow/python/ops/
Dtemplate.py22 from tensorflow.python.ops import variable_scope
314 with variable_scope._pure_variable_scope( # pylint:disable=protected-access
316variable_scope._get_unique_variable_scope(self._name)), # pylint:disable=protected-access
397 with variable_scope.variable_scope(
403 with variable_scope.variable_scope(
420 def variable_scope(self): member in Template
504 default = variable_scope._get_default_variable_store() # pylint: disable=protected-access
506 self._eager_variable_store = variable_scope.EagerVariableStore(default)
510 self._eager_variable_store = variable_scope.EagerVariableStore()
523 default = variable_scope._get_default_variable_store() # pylint: disable=protected-access
[all …]
Dmetrics_impl.py30 from tensorflow.python.ops import variable_scope
72 return variable_scope.variable(
79 synchronization=variable_scope.VariableSynchronization.ON_READ,
80 aggregation=variable_scope.VariableAggregation.SUM,
444 with variable_scope.variable_scope(name, 'mean', (values, weights)):
897 with variable_scope.variable_scope(name, 'auc',
1226 with variable_scope.variable_scope(name, 'mean_accuracy',
1335 with variable_scope.variable_scope(name, 'mean_iou',
1563 with variable_scope.variable_scope(name, 'mean', (values, weights)):
1737 with variable_scope.variable_scope(name, 'false_negatives',
[all …]
/external/tensorflow/tensorflow/python/profiler/internal/
Dmodel_analyzer_testlib.py27 from tensorflow.python.ops import variable_scope
37 _ = variable_scope.get_variable(
41 kernel = variable_scope.get_variable(
46 kernel = variable_scope.get_variable(
58 with variable_scope.variable_scope('inp_%d' % i):
75 kernel1 = variable_scope.get_variable(
81 kernel2 = variable_scope.get_variable(
/external/tensorflow/tensorflow/python/tpu/
Dbfloat16.py22 from tensorflow.python.ops import variable_scope
73 ) -> Generator[variable_scope.variable_scope, None, None]:
86 with variable_scope.variable_scope(
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants_test.py51 from tensorflow.python.ops import variable_scope
951 with variable_scope.variable_scope("", use_resource=use_resource):
952 variable_node = variable_scope.get_variable(
954 variable_scope.get_variable("unused_variable_node", initializer=1.0)
983 with variable_scope.variable_scope("", use_resource=True):
984 variable_node = variable_scope.get_variable(
986 another_variable = variable_scope.get_variable(
1095 with variable_scope.variable_scope("", use_resource=True):
1096 x = variable_scope.get_variable("var_x", initializer=1.0)
1097 y = variable_scope.get_variable("var_y", initializer=2.0)
[all …]
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py33 from tensorflow.python.ops import variable_scope
400 v2 = variable_scope.get_variable(
418 v2 = variable_scope.get_variable(
442 with variable_scope.variable_scope(
443 'reuse', reuse=variable_scope.AUTO_REUSE):
444 v = variable_scope.get_variable(
449 with variable_scope.variable_scope(
450 'reuse', reuse=variable_scope.AUTO_REUSE):
451 v = variable_scope.get_variable(
456 with variable_scope.variable_scope(
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Drnn_cell_test.py46 from tensorflow.python.ops import variable_scope
249 with variable_scope.variable_scope("share_scope"):
251 with variable_scope.variable_scope("drop_scope"):
282 with variable_scope.variable_scope("drop_scope"):
325 with variable_scope.variable_scope(prefix) as scope:
453 with variable_scope.variable_scope("share_scope"):
487 with variable_scope.variable_scope("share_scope"):
545 with variable_scope.variable_scope("share_scope"):
626 with variable_scope.variable_scope("root") as scope:
773 with variable_scope.variable_scope("noshard_scope"):
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
Dmap_fn_test.py33 from tensorflow.python.ops import variable_scope
44 with variable_scope.variable_scope("body"):
46 two = variable_scope.get_variable(
119 with variable_scope.variable_scope("body"):
121 two = variable_scope.get_variable(
127 with variable_scope.variable_scope("root") as varscope:

12345678