Searched refs:apply_indexed_slices_grad (Results 1 – 6 of 6) sorted by relevance
122 accum_op = q.apply_indexed_slices_grad(146 q.apply_indexed_slices_grad(t).run()177 accums[i].apply_indexed_slices_grad(st).run()191 accum_op = q.apply_indexed_slices_grad(grad_indexed_slices)212 accum_op = q.apply_indexed_slices_grad(grad_indexed_slices)239 accum_op = q.apply_indexed_slices_grad(grad_indexed_slices, local_step=0)256 accum_op = q.apply_indexed_slices_grad(grad_indexed_slices, local_step=1)279 accum_ops.append(q.apply_indexed_slices_grad(x, local_step=0))282 def apply_indexed_slices_grad(accum_op): function287 target=apply_indexed_slices_grad, args=(o,)) for o in accum_ops[all …]
27 name: "apply_indexed_slices_grad"
299 train_ops.append(grad_accum.apply_indexed_slices_grad(
1365 def apply_indexed_slices_grad(self, grad, local_step=0, name=None): member in SparseConditionalAccumulator