Home
last modified time | relevance | path

Searched defs:grad1 (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
H A DGraphTest.java214 Output<?>[] grad1 = g.addGradients(null, toArray(y0), toArray(x), toArray(grad0[0])); in addGradientsWithInitialValuesToGraph() local
241 Output<?>[] grad1 = g.addGradients(null, toArray(y0), toArray(x), null); in validateGradientsNames() local
/aosp_15_r20/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/op/core/
H A DGradientsTest.java127 Gradients grad1 = Gradients.create(scope.withName("MyGradients"), y, Arrays.asList(x)); in validateGradientsNames() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dgradient_checker_v2.py345 def max_error(grad1, grad2): argument
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_autograd_function.py429 def backward(ctx, grad1, grad2): argument
1074 def backward(ctx, grad1, grad2): argument
1146 def backward(ctx, grad1, grad2, grad3): argument
/aosp_15_r20/external/dng_sdk/source/
H A Ddng_bad_pixels.cpp824 uint32 grad1; in FixIsolatedPixel() local
1156 uint32 grad1; in FixSingleColumn() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler_test.cc810 auto grad1 = ops::TensorArrayGrad(scope, arg, flow, "grad1"); in TEST_F() local
871 auto grad1 = ops::TensorArrayGrad(scope, arg, flow, "grad1"); in TEST_F() local
903 auto grad1 = ops::TensorArrayGrad(scope, arg, flow, "grad2"); in TEST_F() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_custom_ops.py1448 def foo_backward(ctx, saved, grad0, grad1): argument
H A Dtest_autograd.py3405 def backward(ctx, grad1, grad2): argument
3426 def backward(ctx, grad1, grad2): argument