Search Options

Results per page
Sort
Preferred Languages
Advance

Results 131 - 140 of 142 for var_handle_op (0.29 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

        to input arguments and outputs of the function(s).
    
        Two types of resources are supported:
        (1) A function argument of TF::ResourceType type (this pass).
        (2) A VarHandleOp in the function (tf-promote-var-handles-to-args).
    
        After the pass,
    
         . The function will have an input argument for each resource that is
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir

        %island = tf_executor.island {
        // expected-remark@above {{ID: 10}}
        // expected-remark@above {{Successors: {11}}}
          %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
          // expected-remark@above {{ID: 0}}
          %vh1 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
          // expected-remark@above {{ID: 1}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 20 04:39:18 UTC 2023
    - 129.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md

       if the variable is read only.
    
       For example, the following pseudo MLIR code (types are left out for
       brevity)
       ```mlir
         func.func @hoist_loop_invariant(%arg0, %arg1) {
    %var = "tf.VarHandleOp"() {container="", shared_name="var_name", device = "/device:CPU:0"}
           %results:2 = "tf.WhileRegion"(%arg0, %arg1) ({
           ^bb0(%arg2, %arg3):
             %0 = "tf.OpA"() {is_stateless = true}
             "tf.Yield"(%0)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 02 02:26:39 UTC 2023
    - 96.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/passes.h

    // aliasing output arguments.
    std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteResourcesToArgsPass(
        llvm::ArrayRef<std::string> functions = {});
    
    // Creates a pass that promotes tf.VarHandleOp to resource arguments for all
    // functions.
    std::unique_ptr<OperationPass<ModuleOp>> CreatePromoteVarHandlesToArgsPass();
    
    // Creates a pass that converts readonly reference variables to the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 31.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/tests/tpu-dynamic-layout-pass.mlir

        tf_device.return %1#0, %1#1 : tensor<!tf_type.string>, tensor<2x!tf_type.string>
      }) {device = "/device:CPU:0"} : () -> (tensor<!tf_type.string>, tensor<2x!tf_type.string>)
      %var = "tf.VarHandleOp"() {container = "c", shared_name = "v", device = "/device:TPU:0"} : () -> tensor<!tf_type.resource<tensor<3x3x1x32xf32>>>
      // CHECK-NOT: "tf.TPUGetLayoutOp"
      // CHECK-NOT: "tf.TPUCopyWithLayout"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:59:10 UTC 2023
    - 29.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      return VerifyUnsortedSegmentReduction(*this);
    }
    
    //===----------------------------------------------------------------------===//
    // VarHandleOp
    //===----------------------------------------------------------------------===//
    
    LogicalResult VarHandleOp::verify() {
      // VarHandleOp requires the resource handle supply a single subtype from
      // which to derive the dtype and shape attributes.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tfrt/tests/mlrt/while_to_map_fn.mlir

      %outputs_8  =  "tf.Const"() {device = "/job:localhost/replica:0/task:0/device:CPU:0", value = dense<1> : tensor<i32>} : () -> tensor<i32>
      // CHECK: [[elems:%.*]] = "tf.VarHandleOp"
      %outputs_10  =  "tf.VarHandleOp"() {_xla_inferred_shapes = [#tf_type.shape<>], allowed_devices = [], container = "", device = "/job:localhost/replica:0/task:0/device:CPU:0", shared_name = "w"} : () -> tensor<!tf_type.resource<tensor<3x1xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 23 06:40:22 UTC 2024
    - 68.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir

      func.return
    }
    
    // -----
    
    // Valid VarHandleOp operation.
    // CHECK-LABEL: func @testVarHandleOp
    func.func @testVarHandleOp() -> tensor<!tf_type.resource<tensor<*xf32>>> {
      %0 = "tf.VarHandleOp"() {
        container = "",
        shared_name = "cd2c89b7-88b7-44c8-ad83-06c2a9158347"
      } : () -> tensor<!tf_type.resource<tensor<*xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 23 14:40:35 UTC 2023
    - 236.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir

      func.return %0 : tensor<*xf32>
    }
    
    // CHECK-LABEL: @erase_tf_var_is_initialized
    func.func @erase_tf_var_is_initialized(%arg0 : tensor<!tf_type.resource<tensor<f32>>>) -> tensor<i1> {
      %vh = "tf.VarHandleOp"() {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf_type.resource<tensor<f32>>>
      %is = "tf.VarIsInitializedOp"(%vh) : (tensor<!tf_type.resource<tensor<f32>>>) -> tensor<i1>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 132.1K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        if (std::optional<absl::string_view> cluster_name =
                GetXlaClusterForNode(*n)) {
          n->set_name(absl::StrCat(*cluster_name, "/", n->name()));
        } else if (n->type_string() == "VarHandleOp") {
          n->set_name(absl::StrCat("varhandle/", n->name()));
        } else {
          // There is room for improvement here.  In particular, it may help to
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top