Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for TensorScatterUpdate (0.24 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir

        // CHECK-NOT: BroadcastTo
        // CHECK: TensorScatterUpdate
        "tf.ResourceScatterUpdate"(%resource, %indices, %updates) {device = ""} : (tensor<*x!tf_type.resource>, tensor<?xi32>, tensor<i32>) -> ()
        tf_device.return
      }) : () -> ()
      func.return
    }
    
    // Test that composite tf.ResourceScatterUpdate operation is decomposed to tf.TensorScatterUpdate when update is unranked
    // CHECK-LABEL: @decompose_unranked_updates
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 51.3K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. 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)
  5. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

                                          nudged_float_min);
    
        rewriter.replaceOp(op, {output});
        return success();
      }
    };
    
    // Lowers InvertPermutation op to TensorScatterUpdate op.
    //
    // Example:
    //
    //   %x = "tf.Const"() {value = dense<[3, 4, 0, 1, 2]> : tensor<5xi32>}
    //   "tf.InvertPermutation"(%x) : (tensor<5xi32>) -> tensor<5xi32>
    //
    // is lowered to
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      // CHECK:  %[[ADD:.*]] = "tf.Add"(%arg2, %[[GATHER]]) : (tensor<1xi64>, tensor<1xi64>) -> tensor<1xi64>
      // CHECK:  %[[SCATTER:.*]] = "tf.TensorScatterUpdate"(%arg0, %arg1, %[[ADD]]) : (tensor<7xi64>, tensor<1x1xi32>, tensor<1xi64>) -> tensor<7xi64>
      // CHECK:  return %[[SCATTER]] : tensor<7xi64>
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK-SAME: unique_indices = false
      // CHECK:  ^bb0(%arg3: tensor<f32>, %arg4: tensor<f32>):
      // CHECK:    mhlo.return %arg4 : tensor<f32>
      // CHECK:  })
      %0 = "tf.TensorScatterUpdate"(%tensor, %indices, %updates) : (tensor<?x?x?xf32>, tensor<?x2xi32>, tensor<?x?xf32>) -> tensor<?x?x?xf32>
      func.return %0 : tensor<?x?x?xf32>
    }
    
    // -----
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

                                         hlo::kInfinityMax, &rewriter);
      }
    
      static ComparisonDirection GetDirection() { return ComparisonDirection::LE; }
    };
    
    // Converts TF TensorScatterUpdate/Min/Max/Add/Sub op into Scatter Op with
    // assignment:
    //
    //   %result = "mhlo.scatter"(%tensor, %indices, %updates)
    //     { dimensions = ... }
    //
    template <typename Derived, typename OpTy>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
      TF_DerivedOperandTypeAttr Tindices = TF_DerivedOperandTypeAttr<1>;
    }
    
    def TF_TensorScatterUpdateOp : TF_Op<"TensorScatterUpdate", [Pure]> {
      let summary = [{
    Scatter `updates` into an existing tensor according to `indices`.
      }];
    
      let description = [{
    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