Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for read_variable (0.26 sec)

  1. tensorflow/compiler/mlir/lite/tests/legalize-tf-variables.mlir

      // CHECK: %[[VAR_VAL:.*]] = "tfl.read_variable"(%[[RESOURCE]]) : (tensor<!tf_type.resource<tensor<1x10xf32>>>) -> tensor<1x10xf32>
      // CHECK: %[[ADD:.*]] = tfl.add %[[VAR_VAL]], %arg0 {fused_activation_function = "NONE"} : tensor<1x10xf32>
      // CHECK: "tfl.assign_variable"(%[[RESOURCE]], %[[ADD]]) : (tensor<!tf_type.resource<tensor<1x10xf32>>>, tensor<1x10xf32>) -> ()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 7.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc

    // resource variables.
    //
    // It converts (VariableV2 -> Identity) to (VarHandle -> ReadVariable).
    //
    // For the background, this pass is a part of hoisting VariableV2 ops by
    // re-using the pipeline for hoisting (VarHandle -> ReadVariable) cases, which
    //  can be done by the following passes:
    //  - Capturing resource values into global tensors (importing saved model).
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/saved_model/core/ops/variable_ops.cc

      TF_RETURN_IF_ERROR(assign_op->AddInput(value));
    
      int num_retvals = 0;
      TF_RETURN_IF_ERROR(assign_op->Execute({}, &num_retvals));
      return Status();
    }
    
    Status ReadVariable(ImmediateExecutionContext* ctx,
                        ImmediateExecutionTensorHandle* variable_handle,
                        DataType dtype, ImmediateTensorHandlePtr* output) {
      ImmediateOpPtr read_op(ctx->CreateOperation());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 11:28:19 UTC 2024
    - 5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/common/tfl_pass_config.h

      bool shape_inference = true;
      // Whether to do TFLite runtime verification.
      bool runtime_verification = true;
      // Whether to enable TFLite variables or not, this will allow
      // mutable variables and produce ReadVariable/AssignVariable ops in TFLite.
      bool enable_tflite_variables = false;
      // Whether to disable the variable freezing pass or not.
      // By default we freeze all variables and disallow mutable variables. When
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 19:05:30 UTC 2024
    - 6.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc

      } else {
        // Typically a tf.VarHandle op.
        return v.getDefiningOp()->emitOpError("Non constant predecessor");
      }
    }
    
    // For the "resource" attribute in a ReadVariable or AssignVariable op,
    // determine the symbol reference to the (new) ml_program::GlobalOp.
    SymbolRefAttr lookupGlobalTensor(func::FuncOp func, Value resource,
                                     SymbolTable &syms,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.6K bytes
    - Viewed (0)
Back to top