Searched refs:apply_grad (Results 1 – 9 of 9) sorted by relevance
102 accum_op = q.apply_grad((10.0,))117 q.apply_grad((e,)).run()137 accums[i].apply_grad((i + 10.0,)).run()152 accum_ops = [q.apply_grad(x) for x in elems]171 q.apply_grad([[1.0, 2.0], [3.0, 4.0]])174 q.apply_grad([[1.0], [2.0], [3.0]])184 accum_op = q.apply_grad(x)210 accum_op = q.apply_grad(x)226 accum_op = q.apply_grad((10.0,))238 accum_op = q.apply_grad((10.0,))[all …]
193 accum_op = q.apply_grad([0, 2],214 accum_op = q.apply_grad([0, 2],241 accum_op = q.apply_grad(258 accum_op = q.apply_grad(442 q.apply_grad(454 q.apply_grad(465 q.apply_grad(478 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)498 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)516 q.apply_grad(grad_indices=0, grad_values=[1.0], grad_shape=[]).run()[all …]
23 name: "apply_grad"
240 var_accum.apply_grad(var._ref(), local_step=global_step))
290 train_ops.append(grad_accum.apply_grad(
1275 def apply_grad(self, grad, local_step=0, name=None): member in ConditionalAccumulator1382 return self.apply_grad(1389 def apply_grad(self, member in SparseConditionalAccumulator