- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TensorScatterUpdate (0.77 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/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/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) -
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/jit/mark_for_compilation_pass.cc
"TensorListSetItem", "TensorListSplit", "TensorListStack", "TensorScatterAdd", "TensorScatterMax", "TensorScatterMin", "TensorScatterSub", "TensorScatterUpdate", "ToBool", "TridiagonalSolve", "TridiagonalMatMul", "TruncatedNormal", "UniformDequantize", "UniformQuantize", "UniformQuantizedAdd",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)