Searched refs:addGradients (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
| D | GraphTest.java | 144 Output<?>[] grads0 = g.addGradients(y1, toArray(x1)); in addGradientsToGraph()
149 Output<?>[] grads1 = g.addGradients(y2, toArray(x1, x2)); in addGradientsToGraph()
183 Output<?>[] grad = g.addGradients(null, toArray(y0, y1), toArray(x), null); in addGradientSumsToGraph()
209 Output<?>[] grad0 = g.addGradients(y1, toArray(y0)); in addGradientsWithInitialValuesToGraph()
214 Output<?>[] grad1 = g.addGradients(null, toArray(y0), toArray(x), toArray(grad0[0])); in addGradientsWithInitialValuesToGraph()
238 Output<?>[] grad0 = g.addGradients(null, toArray(y0), toArray(x), null); in validateGradientsNames()
241 Output<?>[] grad1 = g.addGradients(null, toArray(y0), toArray(x), null); in validateGradientsNames()
244 Output<?>[] grad2 = g.addGradients("more_gradients", toArray(y0), toArray(x), null); in validateGradientsNames()
247 Output<?>[] grad3 = g.addGradients("even_more_gradients", toArray(y0), toArray(x), null); in validateGradientsNames()
251 g.addGradients("even_more_gradients", toArray(y0), toArray(x), null); in validateGradientsNames()
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
| D | Graph.java | 169 public Output<?>[] addGradients(String prefix, Output<?>[] y, Output<?>[] x, Output<?>[] dx) { in addGradients() method in Graph
202 addGradients( in addGradients()
235 public Output<?>[] addGradients(Output<?> y, Output<?>[] x) { in addGradients() method in Graph
236 return addGradients(null, new Output<?>[] {y}, x, null); in addGradients()
459 private static native long[] addGradients( in addGradients() method in Graph
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/op/core/ |
| D | Gradients.java | 97 .addGradients( in create()
|