Searched refs:apply_grad (Results 1 – 7 of 7) sorted by relevance
67 accum_op = q.apply_grad((10.0,))82 q.apply_grad((e,)).run()102 accums[i].apply_grad((i + 10.0,)).run()117 accum_ops = [q.apply_grad(x) for x in elems]136 q.apply_grad([[1.0, 2.0], [3.0, 4.0]])139 q.apply_grad([[1.0], [2.0], [3.0]])149 accum_op = q.apply_grad(x)175 accum_op = q.apply_grad(x)191 accum_op = q.apply_grad((10.0,))203 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(457 q.apply_grad(469 q.apply_grad(480 q.apply_grad(493 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)513 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)531 q.apply_grad(grad_indices=0, grad_values=[1.0], grad_shape=[]).run()[all …]
23 name: "apply_grad"
288 train_ops.append(grad_accum.apply_grad(
1358 def apply_grad(self, grad, local_step=0, name=None): member in ConditionalAccumulator1464 return self.apply_grad(1471 def apply_grad(self, member in SparseConditionalAccumulator