Searched refs:grad_op (Results 1 – 3 of 3) sorted by relevance
30 auto grad_op = Roll(scope, grad_inputs[0], Neg(scope, shift), axis); in RollGrad() local31 grad_outputs->push_back(grad_op); in RollGrad()
50 auto grad_op = Unstack(scope, grad_inputs[0], N, Unstack::Axis(axis)); in PackGrad() local51 for (const Output& o : grad_op.output) { in PackGrad()
280 grad_op = g_r.op.inputs[0].op281 self.assertTrue(grad_op.get_attr("_XlaCompile"))282 self.assertEqual(b"jit_scope_0", grad_op.get_attr("_XlaScope"))309 grad_op = g_r.op.inputs[0].op310 self.assertTrue(grad_op.get_attr("_XlaCompile"))312 grad_op.get_attr("_XlaScope"))