Home
last modified time | relevance | path

Searched defs:batch_jacobian (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients.py83 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): function
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape.py193 def batch_jacobian(self, member in LossScaleGradientTape
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py1206 def batch_jacobian(self, member in GradientTape