Searched refs:supports_default_grad (Results 1 – 3 of 3) sorted by relevance
68 def supports_default_grad(t): function
671 elif default_gradient.supports_default_grad(op.outputs[i]):
541 default_gradient.supports_default_grad(while_op_input) and grad is None):