Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TensorScatterUpdate (0.33 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td

           (TF_ConstOp (GetScalarOfType<-1> $indices))),
          $updates
        ),
        (CreateConstBoolAttrFalse)
      )>;
    
    // Pattern to decompose tf.ResourceScatterUpdate into tf.ReadVariable,
    // tf.TensorScatterUpdate, and tf.AssignVariable.
    def DecomposeResourceScatterUpdate : Pat<
      (TF_ResourceScatterUpdateOp:$src_op $resource, $indices, $updates),
      (TF_AssignVariableOp
        $resource,
        (TF_TensorScatterUpdateOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 20.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td

                              /*incompatible_shape_error*/ConstBoolAttrTrue)>;
    
    //===----------------------------------------------------------------------===//
    // TensorScatterUpdate op patterns.
    //===----------------------------------------------------------------------===//
    
    def LowerTensorScatterUpdate_1 : Pat<(TF_TensorScatterUpdateOp $input, $indices, $updates),
              (TF_CastOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 24.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc

      if (maybe_contiguous_start == 0 && buffer_type == updates_type) {
        return AccumulateBuffers(buffer, updates, builder, loc);
      }
      // We cannot simply use a TensorScatterUpdate, as it does not accumulate with
      // the old data; it is tricky to manually add the old data either, since there
      // could be duplicates in the index. We follow the old bridge's approach by
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.5K bytes
    - Viewed (0)
Back to top