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 D | GraphTest.java | 214 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 D | GradientsTest.java | 127 Gradients grad1 = Gradients.create(scope.withName("MyGradients"), y, Arrays.asList(x)); in validateGradientsNames() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | gradient_checker_v2.py | 345 def max_error(grad1, grad2): argument
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_autograd_function.py | 429 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 D | dng_bad_pixels.cpp | 824 uint32 grad1; in FixIsolatedPixel() local 1156 uint32 grad1; in FixSingleColumn() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | xla_compiler_test.cc | 810 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 D | test_custom_ops.py | 1448 def foo_backward(ctx, saved, grad0, grad1): argument
|
H A D | test_autograd.py | 3405 def backward(ctx, grad1, grad2): argument 3426 def backward(ctx, grad1, grad2): argument
|