Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for var_handle (0.13 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir

        // CHECK: [[OLD_VAR:%.*]] = "tf.ReadVariableOp"([[VAR_HANDLE]]) : (tensor<*x!tf_type.resource<tensor<*xf32>>>) -> tensor<*xf32>
        // CHECK: [[NEW_VAR:%.*]] = "tf.Sub"([[OLD_VAR]], [[VAR_DELTA]])
        // CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[NEW_VAR]])
        // CHECK: "tf.AssignVariableOp"([[M_HANDLE]], [[NEW_M]])
        // CHECK: "tf.AssignVariableOp"([[V_HANDLE]], [[NEW_V]])
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 51.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

                   Node** var_handle_op = nullptr) {
      Output var_handle =
          ops::VarHandleOp(scope.WithOpName("Var" + id), DT_FLOAT, TensorShape({}));
      Output read =
          ops::ReadVariableOp(scope.WithOpName("Read" + id), var_handle, DT_FLOAT);
      if (var_handle_op) {
        *var_handle_op = var_handle.node();
      }
      return read.node();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/convert_control_to_data_outputs.mlir

        %add2:2 = tf_executor.island wraps "tf.Add"(%arg0, %one#0) : (tensor<i32>, tensor<i32>) -> tensor<i32>
        %var_handle:2 = tf_executor.island wraps "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
        %assign = tf_executor.island wraps "tf.AssignVariableOp"(%var_handle, %arg1) : (!tf_res, tensor<f32>) -> ()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 18:35:00 UTC 2024
    - 68.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

                GetXlaClusterForNode(*n)) {
          n->set_name(absl::StrCat(*cluster_name, "/", n->name()));
        } else if (n->type_string() == "VarHandleOp") {
          n->set_name(absl::StrCat("varhandle/", n->name()));
        } else {
          // There is room for improvement here.  In particular, it may help to
          // split these unclustered nodes into classes where every node in a
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top