Searched refs:grad_op (Results 1 – 3 of 3) sorted by relevance
262 grad_op = g_r.op.inputs[0].op263 self.assertTrue(grad_op.get_attr("_XlaCompile"))264 self.assertEqual(b"jit_scope_0", grad_op.get_attr("_XlaScope"))290 grad_op = g_r.op.inputs[0].op291 self.assertTrue(grad_op.get_attr("_XlaCompile"))293 grad_op.get_attr("_XlaScope"))
323 grad_op = outputs[0].op326 util.maybe_set_lowering_attr(grad_op)327 util.maybe_propagate_compile_time_consts_in_xla(grad_op)
51 auto grad_op = Unstack(scope, grad_inputs[0], N, Unstack::Axis(axis)); in PackGrad() local52 for (const Output& o : grad_op.output) { in PackGrad()