Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/contrib/optimizer_v2/
Drmsprop.py119 rms = state.get_slot(var, "rms")
120 mom = state.get_slot(var, "momentum")
122 mg = state.get_slot(var, "mg")
149 rms = state.get_slot(var, "rms")
150 mom = state.get_slot(var, "momentum")
152 mg = state.get_slot(var, "mg")
177 rms = state.get_slot(var, "rms")
178 mom = state.get_slot(var, "momentum")
180 mg = state.get_slot(var, "mg")
207 rms = state.get_slot(var, "rms")
[all …]
Dadadelta.py62 accum = state.get_slot(var, "accum")
63 accum_update = state.get_slot(var, "accum_update")
75 accum = state.get_slot(var, "accum")
76 accum_update = state.get_slot(var, "accum_update")
88 accum = state.get_slot(var, "accum")
89 accum_update = state.get_slot(var, "accum_update")
102 accum = state.get_slot(var, "accum")
103 accum_update = state.get_slot(var, "accum_update")
Drmsprop_test.py119 mg0 = opt.get_slot(var0, "mg")
121 mg1 = opt.get_slot(var1, "mg")
123 rms0 = opt.get_slot(var0, "rms")
125 rms1 = opt.get_slot(var1, "rms")
127 mom0 = opt.get_slot(var0, "momentum")
129 mom1 = opt.get_slot(var1, "momentum")
244 mg0 = opt.get_slot(var0, "mg")
246 mg1 = opt.get_slot(var1, "mg")
248 rms0 = opt.get_slot(var0, "rms")
250 rms1 = opt.get_slot(var1, "rms")
[all …]
Dmomentum.py81 mom = state.get_slot(var, "momentum")
92 mom = state.get_slot(var, "momentum")
103 mom = state.get_slot(var, "momentum")
115 mom = state.get_slot(var, "momentum")
Dadam.py116 m = state.get_slot(var, "m")
117 v = state.get_slot(var, "v")
133 m = state.get_slot(var, "m")
134 v = state.get_slot(var, "v")
159 m = state.get_slot(var, "m")
165 v = state.get_slot(var, "v")
Dadagrad.py84 acc = state.get_slot(var, "accumulator")
93 acc = state.get_slot(var, "accumulator")
102 acc = state.get_slot(var, "accumulator")
112 acc = state.get_slot(var, "accumulator")
/external/tensorflow/tensorflow/python/training/
Drmsprop.py141 rms = self.get_slot(var, "rms")
142 mom = self.get_slot(var, "momentum")
144 mg = self.get_slot(var, "mg")
169 rms = self.get_slot(var, "rms")
170 mom = self.get_slot(var, "momentum")
172 mg = self.get_slot(var, "mg")
197 rms = self.get_slot(var, "rms")
198 mom = self.get_slot(var, "momentum")
200 mg = self.get_slot(var, "mg")
227 rms = self.get_slot(var, "rms")
[all …]
Dadadelta.py82 accum = self.get_slot(var, "accum")
83 accum_update = self.get_slot(var, "accum_update")
95 accum = self.get_slot(var, "accum")
96 accum_update = self.get_slot(var, "accum_update")
108 accum = self.get_slot(var, "accum")
109 accum_update = self.get_slot(var, "accum_update")
122 accum = self.get_slot(var, "accum")
123 accum_update = self.get_slot(var, "accum_update")
Dadagrad_da.py109 g_acc = self.get_slot(var, "gradient_accumulator")
110 gg_acc = self.get_slot(var, "gradient_squared_accumulator")
125 g_acc = self.get_slot(var, "gradient_accumulator")
126 gg_acc = self.get_slot(var, "gradient_squared_accumulator")
141 g_acc = self.get_slot(var, "gradient_accumulator")
142 gg_acc = self.get_slot(var, "gradient_squared_accumulator")
158 g_acc = self.get_slot(var, "gradient_accumulator")
159 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.py148 accum = self.get_slot(var, "accum")
149 linear = self.get_slot(var, "linear")
180 accum = self.get_slot(var, "accum")
181 linear = self.get_slot(var, "linear")
212 accum = self.get_slot(var, "accum")
213 linear = self.get_slot(var, "linear")
246 accum = self.get_slot(var, "accum")
247 linear = self.get_slot(var, "linear")
Dproximal_adagrad.py89 acc = self.get_slot(var, "accumulator")
97 acc = self.get_slot(var, "accumulator")
105 acc = self.get_slot(var, "accumulator")
114 acc = self.get_slot(var, "accumulator")
Dadagrad.py97 acc = self.get_slot(var, "accumulator")
106 acc = self.get_slot(var, "accumulator")
115 acc = self.get_slot(var, "accumulator")
125 acc = self.get_slot(var, "accumulator")
Dmomentum.py96 mom = self.get_slot(var, "momentum")
106 mom = self.get_slot(var, "momentum")
116 mom = self.get_slot(var, "momentum")
126 mom = self.get_slot(var, "momentum")
Dadam.py146 m = self.get_slot(var, "m")
147 v = self.get_slot(var, "v")
163 m = self.get_slot(var, "m")
164 v = self.get_slot(var, "v")
189 m = self.get_slot(var, "m")
195 v = self.get_slot(var, "v")
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Drmsprop.py136 rms = self.get_slot(var, "rms")
141 mom = self.get_slot(var, "momentum")
143 mg = self.get_slot(var, "mg")
171 mg = self.get_slot(var, "mg")
181 rms = self.get_slot(var, "rms")
186 mom = self.get_slot(var, "momentum")
188 mg = self.get_slot(var, "mg")
221 mg = self.get_slot(var, "mg")
Drmsprop_test.py126 mg0 = opt.get_slot(var0, "mg")
127 mg1 = opt.get_slot(var1, "mg")
133 mom0 = opt.get_slot(var0, "momentum")
134 mom1 = opt.get_slot(var1, "momentum")
139 rms0 = opt.get_slot(var0, "rms")
141 rms1 = opt.get_slot(var1, "rms")
206 rms0 = opt.get_slot(var0, "rms")
208 rms1 = opt.get_slot(var1, "rms")
211 mom0 = opt.get_slot(var0, "momentum")
212 mom1 = opt.get_slot(var1, "momentum")
[all …]
Dadam.py167 m = self.get_slot(var, 'm')
168 v = self.get_slot(var, 'v')
189 vhat = self.get_slot(var, 'vhat')
216 m = self.get_slot(var, 'm')
223 v = self.get_slot(var, 'v')
235 v_hat = self.get_slot(var, 'vhat')
Dadadelta.py122 accum_grad = self.get_slot(var, 'accum_grad')
123 accum_var = self.get_slot(var, 'accum_var')
137 accum_grad = self.get_slot(var, 'accum_grad')
138 accum_var = self.get_slot(var, 'accum_var')
Dftrl.py154 accum = self.get_slot(var, 'accumulator')
155 linear = self.get_slot(var, 'linear')
188 accum = self.get_slot(var, 'accumulator')
189 linear = self.get_slot(var, 'linear')
/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")
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dnadam_optimizer.py36 m = self.get_slot(var, "m")
37 v = self.get_slot(var, "v")
54 m = self.get_slot(var, "m")
55 v = self.get_slot(var, "v")
81 m = self.get_slot(var, "m")
89 v = self.get_slot(var, "v")
Dreg_adagrad_optimizer.py68 acc = self.get_slot(var, "accumulator")
78 acc = self.get_slot(var, "accumulator")
88 acc = self.get_slot(var, "accumulator")
99 acc = self.get_slot(var, "accumulator")
Dadamax.py112 m = self.get_slot(var, "m")
113 v = self.get_slot(var, "v")
125 m = self.get_slot(var, "m")
126 v = self.get_slot(var, "v")
146 m = self.get_slot(var, "m")
152 v = self.get_slot(var, "v")
Dadam_gs_optimizer.py146 m = self.get_slot(var, "m")
147 v = self.get_slot(var, "v")
163 m = self.get_slot(var, "m")
164 v = self.get_slot(var, "v")
189 m = self.get_slot(var, "m")
195 v = self.get_slot(var, "v")

12345