Home
last modified time | relevance | path

Searched refs:get_slot (Results 1 – 25 of 97) sorted by relevance

1234

/external/tensorflow/tensorflow/python/training/
Dadadelta.py84 accum = self.get_slot(var, "accum")
85 accum_update = self.get_slot(var, "accum_update")
97 accum = self.get_slot(var, "accum")
98 accum_update = self.get_slot(var, "accum_update")
110 accum = self.get_slot(var, "accum")
111 accum_update = self.get_slot(var, "accum_update")
124 accum = self.get_slot(var, "accum")
125 accum_update = self.get_slot(var, "accum_update")
Drmsprop.py145 rms = self.get_slot(var, "rms")
146 mom = self.get_slot(var, "momentum")
148 mg = self.get_slot(var, "mg")
173 rms = self.get_slot(var, "rms")
174 mom = self.get_slot(var, "momentum")
176 mg = self.get_slot(var, "mg")
201 rms = self.get_slot(var, "rms")
202 mom = self.get_slot(var, "momentum")
204 mg = self.get_slot(var, "mg")
231 rms = self.get_slot(var, "rms")
[all …]
Dadagrad_da.py112 g_acc = self.get_slot(var, "gradient_accumulator")
113 gg_acc = self.get_slot(var, "gradient_squared_accumulator")
128 g_acc = self.get_slot(var, "gradient_accumulator")
129 gg_acc = self.get_slot(var, "gradient_squared_accumulator")
144 g_acc = self.get_slot(var, "gradient_accumulator")
145 gg_acc = self.get_slot(var, "gradient_squared_accumulator")
161 g_acc = self.get_slot(var, "gradient_accumulator")
162 gg_acc = self.get_slot(var, "gradient_squared_accumulator")
Drmsprop_test.py122 mg0 = opt.get_slot(var0, "mg")
124 mg1 = opt.get_slot(var1, "mg")
126 rms0 = opt.get_slot(var0, "rms")
128 rms1 = opt.get_slot(var1, "rms")
130 mom0 = opt.get_slot(var0, "momentum")
132 mom1 = opt.get_slot(var1, "momentum")
247 mg0 = opt.get_slot(var0, "mg")
249 mg1 = opt.get_slot(var1, "mg")
251 rms0 = opt.get_slot(var0, "rms")
253 rms1 = opt.get_slot(var1, "rms")
[all …]
Dftrl.py163 accum = self.get_slot(var, "accum")
164 linear = self.get_slot(var, "linear")
195 accum = self.get_slot(var, "accum")
196 linear = self.get_slot(var, "linear")
227 accum = self.get_slot(var, "accum")
228 linear = self.get_slot(var, "linear")
261 accum = self.get_slot(var, "accum")
262 linear = self.get_slot(var, "linear")
Dproximal_adagrad.py95 acc = self.get_slot(var, "accumulator")
103 acc = self.get_slot(var, "accumulator")
111 acc = self.get_slot(var, "accumulator")
120 acc = self.get_slot(var, "accumulator")
Dmomentum.py101 mom = self.get_slot(var, "momentum")
111 mom = self.get_slot(var, "momentum")
121 mom = self.get_slot(var, "momentum")
131 mom = self.get_slot(var, "momentum")
Dadagrad.py98 acc = self.get_slot(var, "accumulator")
107 acc = self.get_slot(var, "accumulator")
116 acc = self.get_slot(var, "accumulator")
126 acc = self.get_slot(var, "accumulator")
Dadam.py153 m = self.get_slot(var, "m")
154 v = self.get_slot(var, "v")
170 m = self.get_slot(var, "m")
171 v = self.get_slot(var, "v")
196 m = self.get_slot(var, "m")
202 v = self.get_slot(var, "v")
Dadadelta_test.py88 slot[0] = adadelta_opt.get_slot(var0, "accum")
92 slot_update[0] = adadelta_opt.get_slot(var0, "accum_update")
96 slot[1] = adadelta_opt.get_slot(var1, "accum")
100 slot_update[1] = adadelta_opt.get_slot(var1, "accum_update")
Dmomentum_test.py78 slot0 = mom_opt.get_slot(var0, "momentum")
80 slot1 = mom_opt.get_slot(var1, "momentum")
302 slot0 = mom_opt.get_slot(var0, "momentum")
305 slot1 = mom_opt.get_slot(var1, "momentum")
481 slot0 = mom_opt.get_slot(var0, "momentum")
483 slot1 = mom_opt.get_slot(var1, "momentum")
556 slot0 = mom_opt.get_slot(var0, "momentum")
558 slot1 = mom_opt.get_slot(var1, "momentum")
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dadam.py171 m = self.get_slot(var, 'm')
172 v = self.get_slot(var, 'v')
188 vhat = self.get_slot(var, 'vhat')
209 m = self.get_slot(var, 'm')
217 v = self.get_slot(var, 'v')
231 v_hat = self.get_slot(var, 'vhat')
425 m = self.get_slot(var, 'm')
426 v = self.get_slot(var, 'v')
434 vhat = self.get_slot(var, 'vhat')
447 m = self.get_slot(var, 'm')
[all …]
Drmsprop.py181 rms = self.get_slot(var, "rms")
183 mom = self.get_slot(var, "momentum")
185 mg = self.get_slot(var, "mg")
214 mg = self.get_slot(var, "mg")
227 rms = self.get_slot(var, "rms")
229 mom = self.get_slot(var, "momentum")
231 mg = self.get_slot(var, "mg")
265 mg = self.get_slot(var, "mg")
Drmsprop_test.py129 mg0 = opt.get_slot(var0, "mg")
130 mg1 = opt.get_slot(var1, "mg")
136 mom0 = opt.get_slot(var0, "momentum")
137 mom1 = opt.get_slot(var1, "momentum")
142 rms0 = opt.get_slot(var0, "rms")
144 rms1 = opt.get_slot(var1, "rms")
210 rms0 = opt.get_slot(var0, "rms")
212 rms1 = opt.get_slot(var1, "rms")
215 mom0 = opt.get_slot(var0, "momentum")
216 mom1 = opt.get_slot(var1, "momentum")
[all …]
Dadadelta.py122 accum_grad = self.get_slot(var, 'accum_grad')
123 accum_var = self.get_slot(var, 'accum_var')
139 accum_grad = self.get_slot(var, 'accum_grad')
140 accum_var = self.get_slot(var, 'accum_var')
Dadamax.py138 m = self.get_slot(var, 'm')
139 v = self.get_slot(var, 'v')
158 m = self.get_slot(var, 'm')
166 v = self.get_slot(var, 'v')
Dftrl.py164 accum = self.get_slot(var, 'accumulator')
165 linear = self.get_slot(var, 'linear')
202 accum = self.get_slot(var, 'accumulator')
203 linear = self.get_slot(var, 'linear')
Dadadelta_test.py82 slot[0] = adadelta_opt.get_slot(var0, "accum_grad")
85 slot_update[0] = adadelta_opt.get_slot(var0, "accum_var")
88 slot[1] = adadelta_opt.get_slot(var1, "accum_grad")
91 slot_update[1] = adadelta_opt.get_slot(var1, "accum_var")
Dnadam.py155 m = self.get_slot(var, 'm')
156 v = self.get_slot(var, 'v')
178 m = self.get_slot(var, 'm')
179 v = self.get_slot(var, 'v')
/external/tensorflow/tensorflow/compiler/tests/
Drmsprop_test.py81 mg0 = rms_opt.get_slot(var0, "mg")
83 mg1 = rms_opt.get_slot(var1, "mg")
85 rms0 = rms_opt.get_slot(var0, "rms")
87 rms1 = rms_opt.get_slot(var1, "rms")
89 mom0 = rms_opt.get_slot(var0, "momentum")
91 mom1 = rms_opt.get_slot(var1, "momentum")
Dadadelta_test.py78 slot[0] = adadelta_opt.get_slot(var0, "accum")
82 slot_update[0] = adadelta_opt.get_slot(var0, "accum_update")
86 slot[1] = adadelta_opt.get_slot(var1, "accum")
90 slot_update[1] = adadelta_opt.get_slot(var1, "accum_update")
Dmomentum_test.py56 slot0 = mom_opt.get_slot(var0, "momentum")
59 slot1 = mom_opt.get_slot(var1, "momentum")
142 slot0 = mom_opt.get_slot(var0, "momentum")
145 slot1 = mom_opt.get_slot(var1, "momentum")
/external/tensorflow/tensorflow/python/training/tracking/
Dutil_with_v1_optimizers_test.py63 self.evaluate(state_ops.assign(optimizer.get_slot(name="m", var=root.var),
87 new_root.optimizer.get_slot(name="m", var=new_root.var)))
89 self.assertIs(new_root.optimizer.get_slot(name="m", var=new_root.var),
99 new_root.optimizer.get_slot(name="m", var=new_root.var)))
155 self.evaluate(optimizer.get_slot(
170 self.evaluate(optimizer.get_slot(
176 self.evaluate(optimizer.get_slot(
183 self.assertAllEqual([5.], self.evaluate(optimizer.get_slot(
191 self.assertAllEqual([2.], self.evaluate(optimizer.get_slot(
/external/mesa3d/src/compiler/nir/
Dnir_lower_io_to_vector.c42 get_slot(const nir_variable *var) in get_slot() function
218 old_vars[get_slot(var)][frac] = var; in create_new_io_vars()
443 const unsigned loc = get_slot(old_var); in nir_lower_io_to_vector_impl()
466 shader, &b, new_var, old_deref, loc - get_slot(new_var)); in nir_lower_io_to_vector_impl()
468 assert(get_slot(new_var) == loc); in nir_lower_io_to_vector_impl()
498 const unsigned loc = get_slot(old_var); in nir_lower_io_to_vector_impl()
513 shader, &b, new_var, old_deref, loc - get_slot(new_var)); in nir_lower_io_to_vector_impl()
515 assert(get_slot(new_var) == loc); in nir_lower_io_to_vector_impl()
/external/tensorflow/tensorflow/python/keras/tests/
Dsaver_test.py80 self.evaluate(optimizer.get_slot(
89 root_trackable.optimizer.get_slot(
99 root_trackable.optimizer.get_slot(

1234