- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for TensorScatterUpdate (0.24 sec)
-
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) -
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 59.8K bytes - Viewed (0) -
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) -
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) -
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)