Searched defs:batch_jacobian (Results 1 – 2 of 2) sorted by relevance
79 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): function
1244 def batch_jacobian(self, member in GradientTape