- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for InplaceUpdate (0.3 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-inplaceupdate.mlir
func.func @tfInplaceUpdate(%arg0: tensor<2x1x2xf32>) -> tensor<2x1x2xf32> { %1 = arith.constant dense<1> : tensor<1xi32> %2 = arith.constant dense<2.0> : tensor<1x1x2xf32> %3 = "tf.InplaceUpdate"(%arg0, %1, %2) {device = ""} : (tensor<2x1x2xf32>, tensor<1xi32>, tensor<1x1x2xf32>) -> tensor<2x1x2xf32> func.return %3 : tensor<2x1x2xf32> } } // CHECK-LABEL: @tfInplaceUpdate
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 16 05:09:09 UTC 2022 - 993 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/tf-tfl-translate-serialize-stablehlo.mlir
module { func.func @tfInplaceUpdate(%arg0: tensor<2x1x2xf32>) -> tensor<2x1x2xf32> { %1 = arith.constant dense<1> : tensor<1xi32> %2 = arith.constant dense<2.0> : tensor<1x1x2xf32> %3 = "tf.InplaceUpdate"(%arg0, %1, %2) {device = ""} : (tensor<2x1x2xf32>, tensor<1xi32>, tensor<1x1x2xf32>) -> tensor<2x1x2xf32> func.return %3 : tensor<2x1x2xf32> } } //CHECK: module attributes
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 18:33:43 UTC 2024 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/tf-tfl-translate-tf-quantize.mlir
module { func.func @tfInplaceUpdate(%arg0: tensor<2x1x2xf32>) -> tensor<2x1x2xf32> { %1 = arith.constant dense<1> : tensor<1xi32> %2 = arith.constant dense<2.0> : tensor<1x1x2xf32> %3 = "tf.InplaceUpdate"(%arg0, %1, %2) {device = ""} : (tensor<2x1x2xf32>, tensor<1xi32>, tensor<1x1x2xf32>) -> tensor<2x1x2xf32> func.return %3 : tensor<2x1x2xf32> } } //CHECK: module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 18:33:43 UTC 2024 - 1.1K bytes - Viewed (0) -
tensorflow/cc/framework/fuzzing/op_fuzzing.bzl
"FakeQuantWithMinMaxVarsPerChannelGradient", "Fill", "Fingerprint", "Gather", "GuaranteeConst", "Identity", "IdentityN", "InplaceAdd", "InplaceSub", "InplaceUpdate", "InvertPermutation", "ListDiff", "MatrixBandPart", "MatrixDiag", "MatrixDiagPart", "MatrixDiagPartV2", "MatrixDiagPartV3", "MatrixDiagV2", "MatrixDiagV3",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 07 19:14:57 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/jit/clone_constants_for_better_clustering.cc
absl::StatusOr<bool> IsSmallConstant(Node* n) { if (!n->IsConstant()) { return false; } return IsConstantSmall(n); } bool IsInPlaceOp(absl::string_view op_name) { return op_name == "InplaceUpdate" || op_name == "InplaceAdd" || op_name == "InplaceSub"; } } // namespace Status CloneConstantsForBetterClusteringPassImpl::CloneSmallConstantInputs(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK-DAG: [[RESHAPE1:%.+]] = mhlo.reshape [[SLICE1]] // CHECK-DAG: [[UPDATE:%.+]] = mhlo.dynamic_update_slice %arg0, [[SLICE2]], [[RESHAPE1]], [[CST]] %0 = "tf.InplaceUpdate"(%arg0, %arg2, %arg1) : (tensor<8x4xf32>, tensor<1xi32>, tensor<1x4xf32>) -> tensor<8x4xf32> // CHECK: return [[UPDATE]] func.return %0 : tensor<8x4xf32> } // -----
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
custom_call->setAttr(kShardingAttr, op.get_XlaShardingAttr()); rewriter.replaceOp(op, custom_call.getResult(0)); return success(); } }; // Converts a TF InplaceUpdate op to DynamicUpdateSlice HLO. class ConvertInplaceUpdateOp : public OpRewritePattern<TF::InplaceUpdateOp> { public: using OpRewritePattern::OpRewritePattern;
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>; } def TF_InplaceUpdateOp : TF_Op<"InplaceUpdate", [Pure]> { let summary = "Updates specified rows 'i' with values 'v'."; let description = [{ Computes `x[i, :] = v; return x`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)