- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for AssignSubVariableOp (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_cluster_formation.mlir
// CHECK-NEXT: [[CONST:%.*]] = "tf.Const" %0 = "tf.Const"() {value = dense<1.000000e+00> : tensor<f32>} : () -> tensor<f32> // CHECK-NEXT: "tf.AssignSubVariableOp"([[UNUSED_RESOURCE]], [[CONST]]) // CHECK: "tf_device.cluster" // CHECK-NEXT: "tf.ReadVariableOp"([[USED_RESOURCE]]) // CHECK-NEXT: "tf.NoOp" // CHECK-NEXT: tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 53.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
tf_device.return }) : () -> () func.return } // Tests that composite tf.AssignSubVariableOp operation is decomposed using // SubOp. // CHECK-LABEL: func @decompose_assign_sub_variable_op func.func @decompose_assign_sub_variable_op() -> () { "tf_device.cluster"() ({
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/jit/mark_for_compilation_pass.cc
const Cluster& cluster) { Node* n = GetOnlyNodeIn(cluster); if (!n) { return false; } if (n->type_string() != "AssignAddVariableOp" && n->type_string() != "AssignSubVariableOp") { return false; } DataType dtype; if (!TryGetNodeAttr(n->def(), "dtype", &dtype) || !DataTypeIsInteger(dtype)) { return false; } Node* const_input = nullptr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)