Searched refs:read_cotangent (Results 1 – 2 of 2) sorted by relevance
165 def read_cotangent(v): function190 cts_in = map(read_cotangent, eqn.outvars)264 return map(read_cotangent, jaxpr.invars)
182 def read_cotangent(v): function217 cts_in = map(read_cotangent, eqn.outvars)219 cts_in, = map(read_cotangent, eqn.outvars)235 cotangents_out = map(read_cotangent, jaxpr.invars)