Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for scatter_nd_add (0.23 sec)

  1. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      // CHECK:  return %[[ADD0]] : tensor<1x1x6x2xf32>
    }
    
    func.func @scatter_nd_add(%arg0: tensor<7xi64>, %arg1: tensor<1x1xi32>, %arg2: tensor<1xi64>) -> tensor<7xi64> {
      %0 = "tf.TensorScatterAdd"(%arg0, %arg1, %arg2) : (tensor<7xi64>, tensor<1x1xi32>, tensor<1xi64>) -> tensor<7xi64>
      func.return %0 : tensor<7xi64>
    
      // CHECK-LABEL: scatter_nd_add
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    ```python
    ref = tf.Variable([1, 2, 3, 4, 5, 6, 7, 8], use_resource=True)
    indices = tf.constant([[4], [3], [1], [7]])
    updates = tf.constant([9, 10, 11, 12])
    add = tf.scatter_nd_add(ref, indices, updates)
    with tf.Session() as sess:
      print sess.run(add)
    ```
    
    The resulting update to ref would look like this:
    
        [1, 13, 3, 14, 14, 6, 7, 20]
    
    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