Search Options

Results per page
Sort
Preferred Languages
Advance

Results 91 - 100 of 136 for var_handle_op (0.25 sec)

  1. 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)
  2. 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)
  3. 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)
  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/compiler/jit/xla_device_ops.h

                                                                                   \
      REGISTER_KERNEL_BUILDER(                                                     \
          Name("VarHandleOp").Device(DEVICE).HostMemory("resource"), VarHandleOp); \
      REGISTER_KERNEL_BUILDER(                                                     \
          Name("_VarHandlesOp").Device(DEVICE).HostMemory("resources"),            \
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 23 19:28:25 UTC 2021
    - 17.1K bytes
    - Viewed (0)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top