Searched refs:get_slot (Results 1 – 25 of 90) sorted by relevance
1234
145 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 …]
84 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")
112 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")
122 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 …]
154 accum = self.get_slot(var, "accum")155 linear = self.get_slot(var, "linear")186 accum = self.get_slot(var, "accum")187 linear = self.get_slot(var, "linear")218 accum = self.get_slot(var, "accum")219 linear = self.get_slot(var, "linear")252 accum = self.get_slot(var, "accum")253 linear = self.get_slot(var, "linear")
95 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")
101 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")
98 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")
150 m = self.get_slot(var, "m")151 v = self.get_slot(var, "v")167 m = self.get_slot(var, "m")168 v = self.get_slot(var, "v")193 m = self.get_slot(var, "m")199 v = self.get_slot(var, "v")
88 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")
77 slot0 = mom_opt.get_slot(var0, "momentum")79 slot1 = mom_opt.get_slot(var1, "momentum")301 slot0 = mom_opt.get_slot(var0, "momentum")304 slot1 = mom_opt.get_slot(var1, "momentum")484 slot0 = mom_opt.get_slot(var0, "momentum")486 slot1 = mom_opt.get_slot(var1, "momentum")559 slot0 = mom_opt.get_slot(var0, "momentum")561 slot1 = mom_opt.get_slot(var1, "momentum")
81 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")
78 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")
56 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")
149 rms = self.get_slot(var, "rms")151 mom = self.get_slot(var, "momentum")153 mg = self.get_slot(var, "mg")182 mg = self.get_slot(var, "mg")195 rms = self.get_slot(var, "rms")197 mom = self.get_slot(var, "momentum")199 mg = self.get_slot(var, "mg")233 mg = self.get_slot(var, "mg")
129 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")209 rms0 = opt.get_slot(var0, "rms")211 rms1 = opt.get_slot(var1, "rms")214 mom0 = opt.get_slot(var0, "momentum")215 mom1 = opt.get_slot(var1, "momentum")[all …]
197 m = self.get_slot(var, 'm')198 v = self.get_slot(var, 'v')214 vhat = self.get_slot(var, 'vhat')235 m = self.get_slot(var, 'm')243 v = self.get_slot(var, 'v')257 v_hat = self.get_slot(var, 'vhat')
132 accum_grad = self.get_slot(var, 'accum_grad')133 accum_var = self.get_slot(var, 'accum_var')149 accum_grad = self.get_slot(var, 'accum_grad')150 accum_var = self.get_slot(var, 'accum_var')
166 accum = self.get_slot(var, 'accumulator')167 linear = self.get_slot(var, 'linear')198 accum = self.get_slot(var, 'accumulator')199 linear = self.get_slot(var, 'linear')
139 m = self.get_slot(var, 'm')140 v = self.get_slot(var, 'v')160 m = self.get_slot(var, 'm')168 v = self.get_slot(var, 'v')
82 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")
171 m = self.get_slot(var, 'm')172 v = self.get_slot(var, 'v')194 m = self.get_slot(var, 'm')195 v = self.get_slot(var, 'v')
42 get_slot(const nir_variable *var) in get_slot() function218 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()
170 optimizer.get_slot(204 m_bias_slot = optimizer.get_slot(model._named_dense.variables[1], "m")238 on_create_m_bias_slot = on_create_optimizer.get_slot(520 self.evaluate(state_ops.assign(optimizer.get_slot(name="m", var=root.var),544 new_root.optimizer.get_slot(name="m", var=new_root.var)))546 self.assertIs(new_root.optimizer.get_slot(name="m", var=new_root.var),556 new_root.optimizer.get_slot(name="m", var=new_root.var)))612 self.evaluate(optimizer.get_slot(627 self.evaluate(optimizer.get_slot(633 self.evaluate(optimizer.get_slot([all …]
194 def get_slot(self, *args, **kwargs): member in CrossShardOptimizer206 return self._opt.get_slot(*args, **kwargs)