Home
last modified time | relevance | path

Searched defs:tangents (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dforwardprop.py43 def _identity_jvp(attr_tuple, inputs, outputs, tangents): argument
54 def _read_variable_jvp(attr_tuple, inputs, outputs, tangents): argument
70 def _jvp_helper(op_name, attr_tuple, inputs, outputs, tangents): argument
142 def _jvp_helper_wrapper(op_name, attr_tuple, inputs, outputs, tangents, argument
202 tangents, argument
328 def __init__(self, primals, tangents): argument
382 def _watch(self, primals, tangents): argument
449 def _batch_accumulator(cls, primals, tangents): argument
Dforwardprop_util.py30 def __new__(cls, indices=None, tangents=None): argument
Dforwardprop_test.py56 def _jvp(f, primals, tangents): argument
91 def _jvp_batch(f, primal, tangents): argument
145 def _hvp(f, primals, tangents): argument
/external/pytorch/torch/_functorch/_aot_autograd/
Dtraced_function_transforms.py315 primals, tangents, fwd_seed, fwd_base_offset, bwd_seed, bwd_base_offset argument
643 def joint_helper(primals, tangents): argument
784 def joint_fn(primals, tangents): argument
Druntime_wrappers.py1812 def get_types_for_tangents(tangents): argument
/external/pytorch/torch/_functorch/
Dautograd_function.py129 def jvp(ctx, *tangents):
434 def jvp(ctx, *tangents):
438 def jvp_no_context(saved_tensors, tangents): argument
494 def get_tangents_in_dims(input_dims, tangents): argument
Deager_transforms.py80 def _jvp_treespec_compare(primals, tangents): argument
93 def _linearize_treespec_compare(primals, tangents): argument
1828 def jvp_fn(*tangents):
/external/pytorch/torch/autograd/
Dfunctional.py526 def jvp(tangents): argument
/external/pytorch/test/functorch/
Dtest_ops.py210 def simulate_jvp(f, primals, tangents): argument
215 def ref_jvp(f, primals, tangents): argument
327 def _get_jvp_variant(fn, primals, tangents): argument
/external/skia/src/core/
DSkPath.cpp2959 SkTDArray<SkVector>* tangents) { in tangent_cubic()
2987 SkTDArray<SkVector>* tangents) { in tangent_conic()
3014 SkTDArray<SkVector>* tangents) { in tangent_quad()
3040 SkTDArray<SkVector>* tangents) { in tangent_line()
3120 SkTDArray<SkVector> tangents; in contains() local