Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for InplaceAdd (0.22 sec)

  1. tensorflow/compiler/jit/clone_constants_for_better_clustering.cc

        //   SRC -> I
        //   SRC -> V
        //   Const -> Identity
        //   Const -> InplaceAdd [label="x"]
        //   I -> InplaceAdd [label="i"]
        //   V -> InplaceAdd [label="v"]
        //   InplaceAdd -> Identity [style=dotted]
        // }
        //
        // then the value produced by `Identity` is Const+I*V since InplaceAdd
        // modifies the tensor in place.  However, if we clone `Const` and turn the
        // graph into:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/cc/framework/fuzzing/op_fuzzing.bzl

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Nov 07 19:14:57 UTC 2022
    - 4.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/clone_constants_for_better_clustering_test.cc

      {
        Output tr0 = ops::Transpose(on_gpu.WithOpName("tr0"), in0, perm);
        Output tr1 = ops::Transpose(on_gpu.WithOpName("tr1"), in1, perm);
      }
    
      Output in_place_add =
          ops::InplaceAdd(on_cpu.WithOpName("tr0"), perm,
                          ops::Placeholder(on_cpu.WithOpName("i"), DT_INT32), perm);
    
      std::unique_ptr<Graph> result;
      TF_ASSERT_OK(CloneConstantsForBetterClustering(root, &result));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      let results = (outs);
    
      TF_DerivedOperandTypeAttr Tkey = TF_DerivedOperandTypeAttr<1>;
      TF_DerivedOperandTypeAttr Tval = TF_DerivedOperandTypeAttr<2>;
    }
    
    def TF_InplaceAddOp : TF_Op<"InplaceAdd", [Pure, TF_AllTypesMatch<["x", "y"]>]> {
      let summary = "Adds v into specified rows of x.";
    
      let description = [{
    Computes y = x; y[i, :] += v; return y.
      }];
    
      let arguments = (ins
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top