- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for TensorScatterUpdate (0.31 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/tests/convert_tf_xla_op_to_tf_op.mlir
// CHECK: %[[arg1_i64:.*]] = "tf.Cast"(%arg1) <{Truncate = false}> : (tensor<1xi32>) -> tensor<1xi64> // CHECK: %[[tensor_scatter_update:.*]] = "tf.TensorScatterUpdate"(%[[cst]], %[[cst_0]], %[[arg1_i64]]) : (tensor<2xi64>, tensor<1x1xi64>, tensor<1xi64>) -> tensor<2xi64> // CHECK: %[[arg2_i64:.*]] = "tf.Cast"(%arg2) <{Truncate = false}> : (tensor<2xi32>) -> tensor<2xi64>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 3.7K bytes - Viewed (0) -
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/cc/framework/fuzzing/cc_op_fuzz_gen.cc
"QuantizedInstanceNorm", "QuantizedReshape", "ScatterNd", "TensorScatterUpdate"}; // TODO(unda, b/253431636): constrained kernel if (constrained_kernel.find(op_info.op_name) != constrained_kernel.end()) { std::cout << "NOT fuzzing: " << op_info.graph_op_def.name()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 16:26:51 UTC 2024 - 13K 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/tensorflow/tests/tensor_list_ops_decomposition.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
%0 = "tf.TensorScatterUpdate"(%tensor, %indices, %updates) : (tensor<4x4x4xf32>, tensor<i32>, tensor<4x4xf32>) -> tensor<4x4x4xf32> func.return %0 : tensor<4x4x4xf32> } // ----- func.func @tensor_scatter_update(%tensor: tensor<4x4x4xf32>, %indices: tensor<4x2xi32>, %updates: tensor<f32>) -> tensor<4x4x4xf32> { // CHECK: TensorScatterUpdate
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir
// CHECK-LABEL: roll_non_constant_shift // CHECK: "tf.Roll" } func.func @scatter_nd_updates(%arg0: tensor<14xf32>, %arg1: tensor<1x1xi32>, %arg2: tensor<1xf32>) -> tensor<14xf32> { %0 = "tf.TensorScatterUpdate"(%arg0, %arg1, %arg2) : (tensor<14xf32>, tensor<1x1xi32>, tensor<1xf32>) -> tensor<14xf32> func.return %0 : tensor<14xf32> // CHECK-LABEL: scatter_nd_updates
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 92K bytes - Viewed (0)