Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 70 of 76 for varhandle_op (0.51 sec)

  1. tensorflow/c/experimental/ops/gen/cpp/cpp_generator_test.cc

          "IdentityN",  // Variadic input+output
          "SparseSoftmaxCrossEntropyWithLogits",  // 2 outputs
          "AccumulatorApplyGradient",             // 0 outputs
          "VarHandleOp",                          // type, shape, list(string) attrs
          "RestoreV2",  // Variadic output-only, list(type) attr
      };
    
      cpp::CppConfig cpp_config(category, name_space);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 18 17:02:28 UTC 2022
    - 2.9K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/c/experimental/ops/resource_variable_ops.h

    #include "tensorflow/c/eager/abstract_context.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    
    namespace tensorflow {
    namespace ops {
    
    // Creates a handle to a Variable resource.
    Status VarHandleOp(AbstractContext* ctx, AbstractTensorHandle** resource,
                       DataType dtype, const PartialTensorShape shape,
                       const char* container = "", const char* shared_name = "",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 10 19:11:36 UTC 2022
    - 2.5K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc

      static const char* kOpsToSkip[] = {
          "IdentityOp",
          "NoOp",              // b/174596063
          "OptionalHasValue",  // b/173136483
          "OptionalGetValue",  // b/173136483
          "VarHandleOp",       // b/176819198
      };
      for (const char* skip : kOpsToSkip) {
        if (absl::StartsWith(orig_op->op_name(), skip)) return absl::OkStatus();
      }
    
      tf_core_op_expansion_node_counter->GetCell()->IncrementBy(1);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Feb 25 16:22:36 UTC 2024
    - 3.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_duplicate_resource_ops.cc

      auto& island_block = island_op.getBody().front();
      if (++island_block.begin() != --island_block.end()) return nullptr;
    
      Operation* resource_op = &island_block.front();
      if (llvm::isa<TF::VarHandleOp, TF::HashTableOp, TF::HashTableV2Op,
                    TF::MutableHashTableV2Op>(resource_op)) {
        return resource_op;
      }
      return nullptr;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 26 04:26:16 UTC 2023
    - 4.7K bytes
    - Viewed (0)
  10. tensorflow/cc/experimental/libtf/tests/variable_test.cc

      {
        AbstractTensorHandle* var_ptr = nullptr;
        PartialTensorShape scalar_shape;
        TF_EXPECT_OK(
            PartialTensorShape::MakePartialShape<int32_t>({}, 0, &scalar_shape));
        TF_EXPECT_OK(tensorflow::ops::VarHandleOp(ctx_.get(), &var_ptr, DT_FLOAT,
                                                  scalar_shape));
        var.reset(var_ptr);
      }
      // Assign a value.
      auto x = CreateScalarTensor<float, TF_FLOAT>(2.0f);
      TF_EXPECT_OK(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 19 21:44:52 UTC 2023
    - 4.5K bytes
    - Viewed (0)
Back to top