Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for VAR_HANDLE (0.18 sec)

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

        // CHECK: [[ACCUM_NEW_LR:%.*]] = "tf.Mul"([[ACCUM_NEW]], [[LR]])
        // CHECK: [[VAR:%.*]] = "tf.ReadVariableOp"([[VAR_HANDLE]])
        // CHECK: [[VAR_NEW:%.*]] = "tf.Sub"([[VAR]], [[ACCUM_NEW_LR]])
        // CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[VAR_NEW]])
    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/mlir/lite/tests/quantize-variables.mlir

      %0 = "tfl.var_handle"() : () -> tensor<!tf_type.resource>
      %1 = "tfl.dequantize"(%arg0) : (tensor<1x2x1x3x!quant.uniform<i8:f32, 1.0>>) -> tensor<1x2x1x3xf32>
      "tfl.assign_variable"(%0, %1) : (tensor<!tf_type.resource>, tensor<1x2x1x3xf32>) -> ()
      func.return %arg0 : tensor<1x2x1x3x!quant.uniform<i8:f32, 1.0>>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 20.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_distributed_test.cc

      TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      TFE_TensorHandle* var_handle = TestVariable(ctx, 2.0, dev2_name);
      EXPECT_NE(var_handle, nullptr);
      TFE_ContextAsyncWait(ctx, status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      const string function_def = VariableAddFunction();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 15 09:49:45 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_test_util.cc

      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_TensorHandle* var_handle = nullptr;
      int num_retvals = 1;
      TFE_Execute(op, &var_handle, &num_retvals, status);
      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_DeleteOp(op);
      if (TF_GetCode(status) != TF_OK) return nullptr;
      CHECK_EQ(1, num_retvals);
      TF_DeleteStatus(status);
      return var_handle;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/saved_model/core/ops/variable_ops.cc

                                      strlen(ResourceHandle::ANONYMOUS_NAME)));
    
      AbstractTensorHandle* var_handle = nullptr;
      int num_retvals = 1;
      TF_RETURN_IF_ERROR(varhandle_op->Execute(
          absl::MakeSpan(&var_handle, num_retvals), &num_retvals));
      AbstractTensorHandlePtr owned_var_handle(var_handle);
      if (!tensorflow::isa<ImmediateExecutionTensorHandle>(
              owned_var_handle.get())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 11:28:19 UTC 2024
    - 5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tests/legalize-tf-variables.mlir

        func.return %2, %3 : tensor<1x10xf32>, tensor<1x10xi64>
      }
    
      // CHECK: %[[RESOURCE:.*]] = "tfl.var_handle"() <{container = "c", shared_name = "a"}> : () -> tensor<!tf_type.resource<tensor<1x10xf32>>>
      // CHECK: %[[RESOURCE_1:.*]] = "tfl.var_handle"() <{container = "c", shared_name = "b"}> : () -> tensor<!tf_type.resource<tensor<1x10xi64>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 7.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.td

        serialization.
    
        This Op will restore the tensors asynchronously. There are N restored tensors
        corresponding to N $var_handle. Each restored tensor is given a unique name,
        typically a concatenation of `container_name` and `shared_name` from a `var_handle`.
        The runtime shall handle the possibility that the tensors are not ready when requested
        because the tensors are loaded asynchronously.
      }];
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 20:44:15 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

    }
    
    Node* MakeWrite(const Scope& scope, Output value_to_write, const string& id) {
      Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT,
                                           TensorShape({}));
      ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id),
                                      var_handle, value_to_write);
      return assign_op.operation.node();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  9. 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)
  10. 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)
Back to top