Search Options

Results per page
Sort
Preferred Languages
Advance

Results 71 - 80 of 114 for var_handle_op (0.17 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir

    // inlined.
    
    func.func private @simple_callee_var() -> tensor<2xi32>  {
      %cst = "tf.Const"() { value = dense<2> : tensor<2xi32> } : () -> tensor<2xi32>
      %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<!tf_type.resource<tensor<2xi32>>>
      "tf.AssignVariableOp"(%0, %cst) {device = ""} : (tensor<!tf_type.resource<tensor<2xi32>>>, tensor<2xi32>) -> ()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 7.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/breakup-islands.mlir

    // CHECK: }
    
    func.func @unknown_side_effecting_op(%arg0: tensor<32xf32>) -> () {
      tf_executor.graph {
        %island = tf_executor.island {
          %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
          %vh1 = "tf.VarHandleOp"() {container = "c", shared_name = "v1"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:59:10 UTC 2023
    - 28.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/update_control_dependencies.mlir

    func.func @unknown_side_effecting_op(%arg0: tensor<32xf32>) {
      tf_executor.graph {
        %outputs, %control = tf_executor.island wraps "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
        %outputs_0, %control_1 = tf_executor.island wraps "tf.VarHandleOp"() {container = "c", shared_name = "v1"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Nov 03 18:12:49 UTC 2023
    - 25.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/mlir_import_options.h

      bool unconditionally_use_set_output_shapes = false;
    
      // Apply default attributes from the op definition to the loaded op.
      bool add_default_attributes = true;
    
      // If set, promote tf.VarHandleOp to resource arguments for all functions.
      bool lift_variables = true;
    
      // Keeps the variables in initializers before lifting variables (when
      // `lift_variables == true`) or newly adding variable initialization patterns
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 20 13:19:26 UTC 2023
    - 2.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/analysis/resource_value_typed_analyzer.cc

      auto* operation = resource.getDefiningOp();
      if (operation && isa<TF::VarHandleOp>(operation)) {
        mutable_variables_.insert(GetResourceKey(operation));
      }
    }
    
    bool ResourceAnalyzer::IsPotentiallyWritten(Value resource) const {
      assert(IsResource(resource));
      auto* operation = resource.getDefiningOp();
      if (operation && isa<TF::VarHandleOp>(operation))
        return mutable_variables_.contains(GetResourceKey(operation));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 8K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/saved_model/core/ops/variable_ops.h

    #include "tensorflow/core/framework/tensor_shape.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/platform/status.h"
    
    namespace tensorflow {
    namespace internal {
    
    // Executes a VarHandleOp using `ctx`, and fills `handle` with the DT_RESOURCE
    // TensorHandle associated with the variable. This is equivalent to creating an
    // unitialized TF2 tf.Variable.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Aug 30 21:44:45 UTC 2022
    - 3.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/freeze_global_tensors.cc

          Operation *op = *latticeElement->getValue().ops.begin();
          GlobalTensorOp globalTensor = llvm::dyn_cast<GlobalTensorOp>(op);
    
          if (!globalTensor)
            continue;  // happens if the name is e.g. in a VarHandleOp.
    
          if (globalTensor.getIsMutable()) {
            freezeable[val] = false;
            continue;
          }
    
          freezeable[val] = true;
    
          // Verify users are supported kind.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/ops/gen/cpp/golden/testing_ops.h.golden

    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 16 19:04:03 UTC 2023
    - 2.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc

    // Creates the per-device variables that represent the formatting state of each
    // device.
    llvm::SmallVector<TF::VarHandleOp, 4> CreateStateVars(
        const llvm::SmallDenseMap<llvm::StringRef, llvm::SmallVector<StringRef, 4>>&
            devices,
        Location loc, RankedTensorType key_type, OpBuilder* builder) {
      llvm::SmallVector<TF::VarHandleOp, 4> state_vars;
    
      // TODO(b/148913020): Remove this constraint once model parallelism is
      // supported.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

                                   result);
    }
    
    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),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
Back to top