HomeSort by relevance Sort by last modified time
    Searched refs:Gradients (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/op/core/
Gradients.java 42 * Gradients gradients = Gradients.create(scope, Arrays.asList(loss), Arrays.asList(w, b));
45 * ApplyGradientDescent.create(scope, w, alpha, gradients.<Float>dy(0));
46 * ApplyGradientDescent.create(scope, b, alpha, gradients.<Float>dy(1));
50 public class Gradients implements Op, Iterable<Operand<?>> {
53 * Optional attributes for {@link Gradients}
73 * Adds gradients computation ops to the graph according to scope.
79 * @return a new instance of {@code Gradients}
81 public static Gradients create
    [all...]
  /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/op/core/
GradientsTest.java 47 Gradients grads = Gradients.create(scope, y1, Arrays.asList(x, y0));
74 Gradients grads = Gradients.create(scope, Arrays.asList(y0, y1), Arrays.asList(x));
99 Gradients grads0 = Gradients.create(scope, y1, Arrays.asList(y0));
100 Gradients grads1 = Gradients.create(scope, y0, Arrays.asList(x), Gradients.dx(grads0.dy()));
124 Gradients grad0 = Gradients.create(scope, y, Arrays.asList(x))
    [all...]
  /external/tensorflow/tensorflow/go/op/
gradients.go 25 // Gradients adds gradients computation ops to the graph according to scope.
33 func Gradients(scope *Scope, y []tf.Output, x []tf.Output, dx ...tf.Output) (output []tf.Output) {
35 scope.UpdateErr("Gradients", fmt.Errorf("Gradients does not currently support control dependencies (via Scope.WithControlDependencies)."))
39 scope.UpdateErr("Gradients", fmt.Errorf("Gradients does not currently support device annotations (via Scope.WithDevice)."))
44 if output, err = scope.graph.AddGradients(scope.opName("Gradients"), y, x, dx); err != nil {
45 scope.UpdateErr("Gradients", err)
gradients_test.go 36 grads0 := Gradients(s, []tf.Output{y1}, []tf.Output{x1})
48 grads1 := Gradients(sub, []tf.Output{y2}, []tf.Output{x1, x2})
102 grad := Gradients(s, []tf.Output{y0, y1}, []tf.Output{x})
143 grads0 := Gradients(s, []tf.Output{y1}, []tf.Output{y0})
155 grads1 := Gradients(sub, []tf.Output{y0}, []tf.Output{x}, grads0[0])
195 grads0 := Gradients(s, []tf.Output{y0}, []tf.Output{x})
199 if !strings.HasPrefix(grads0[0].Op.Name(), "Gradients/") {
200 t.Fatalf("Got name %v, wanted started with Gradients/", grads0[0].Op.Name())
204 grads1 := Gradients(sub, []tf.Output{y0}, []tf.Output{x})
208 if !strings.HasPrefix(grads1[0].Op.Name(), "sub/Gradients/") {
    [all...]
wrappers.go 64 // Compute gradients for a FakeQuantWithMinMaxVarsPerChannel operation.
67 // gradients: Backpropagated gradients above the FakeQuantWithMinMaxVars operation,
70 // same as `gradients`.
75 // Returns Backpropagated gradients w.r.t. inputs, shape same as
77 // `gradients * (inputs >= min && inputs <= max)`.Backpropagated gradients w.r.t. min parameter, shape `[d]`:
78 // `sum_per_d(gradients * (inputs < min))`.Backpropagated gradients w.r.t. max parameter, shape `[d]`:
79 // `sum_per_d(gradients * (inputs > max))`
    [all...]
  /external/tensorflow/tensorflow/python/ops/
random_grad.py 15 """Gradients for operators defined in random_ops.py."""
40 "Implicit Reparameterization Gradients" (https://arxiv.org/abs/1805.08498).
  /external/swiftshader/third_party/llvm-7.0/llvm/lib/Target/AMDGPU/Utils/
AMDGPUBaseInfo.h 196 bool Gradients;
  /external/skia/fuzz/
FuzzGradients.cpp 253 DEF_FUZZ(Gradients, fuzz) {
  /external/skqp/fuzz/
FuzzGradients.cpp 253 DEF_FUZZ(Gradients, fuzz) {
  /external/tensorflow/tensorflow/c/
while_loop_test.cc 414 TEST_F(CApiWhileLoopTest, Gradients) {
  /external/swiftshader/third_party/llvm-7.0/llvm/lib/Target/AMDGPU/
SIISelLowering.cpp     [all...]

Completed in 438 milliseconds