Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 44 for var_handle_op (0.14 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/promote_resources_to_args.mlir

      // CHECK-NOT: "tf.VarHandleOp"
      // CHECK-NOT: "tf.ReadVariableOp"
      // CHECK: %[[CONST:.*]] = "tf.Const"()
      // CHECK: %[[ADD:[0-9]*]] = "tf.AddV2"(%arg1, %[[CONST]])
      // CHECK: %[[PACK:[0-9]*]] = "tf.Pack"(%[[CONST]], %[[ADD]])
      // CHECK: return %[[PACK]]
      %0 = "tf.Const"() {value = dense<4.200000e+01> : tensor<f32>} : () -> tensor<f32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 18.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/resource-alias-analysis-test.mlir

        // CHECK: tf_executor.island
        %island = tf_executor.island {
          // expected-remark@below {{Result #0, ID 0 : 0, 1, 2}}
          %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
          // expected-remark@below {{Result #0, ID 1 : 0, 1, 2}}
          %vh1 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
          // expected-remark@below {{Result #0, ID 2 : 0, 1, 2}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jul 21 17:19:47 UTC 2023
    - 22.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/tests/hoist_invariant_ops.mlir

      // CHECK-NOT: tf.VarHandleOp
      // CHECK-NOT: tf.ReadVariableOp
      // CHECK: [[v:%.*]] = "tf._TfrtGetResource"() <{container = [""], indices = [0], shared_name = [""]}> {device = "/CPU:0"} : () -> tensor<i32>
      // CHECK: [[r:%.*]] = "tf.AddV2"({{.*}}, [[v]]) {device = "/CPU:0"} : (tensor<i32>, tensor<i32>) -> tensor<i32>
      // CHECK: return [[r]]
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 01 23:54:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_duplicate_resource_ops.mlir

        %outputs_7, %control_8 = tf_executor.island wraps "tf.Const"() {value = dense<"MatMul/b_0"> : tensor<1x!tf_type.string>} : () -> tensor<1x!tf_type.string>
        %outputs_9, %control_10 = tf_executor.island wraps "tf.VarHandleOp"() {container = "", shared_name = "MatMul/b_0"} : () -> tensor<!tf_type.resource<tensor<20x4096xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 26 04:26:16 UTC 2023
    - 10.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/tests/resource-device-inference.mlir

      %arg0: !tf_res {tf.device = "/TPU:0"},
      %arg1: !tf_res {tf.device = "/TPU:1"}) {
      tf_executor.graph {
        // CHECK: tf_executor.island
        %island = tf_executor.island {
          // CHECK-NEXT: "tf.VarHandleOp"
          %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0", device = "/CPU:0"}
            : () -> !tf_res
          // CHECK-NEXT: "tf.Identity"
          // CHECK-SAME: {device = "/TPU:0"}
          %id0 = "tf.Identity"(%arg0) : (!tf_res)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 17 16:01:45 UTC 2022
    - 18.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/tests/hoist_loop_invariant.mlir

    // readonly.
    
    // CHECK-LABEL: readvariableop_is_hoisted_if_readonly2
    // CHECK:       [[CST_0:%.*]] = "tf.Const"
    // CHECK:       [[VAR_1:%.*]] = "tf.VarHandleOp"
    // CHECK-SAME:  "shared_name_var1"
    // CHECK:       [[VAR_2:%.*]] = "tf.VarHandleOp"
    // CHECK-SAME:  "shared_name_var2"
    // CHECK:       [[CST_1:%.*]] = "tf.Const"
    // CHECK:       [[VAR_VAL:%.*]] = "tf.ReadVariableOp"([[VAR_1]])
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Aug 22 17:12:02 UTC 2023
    - 14.2K bytes
    - Viewed (0)
  7. tensorflow/cc/tools/freeze_saved_model.cc

    }
    
    // Returns the name of the VarHandleOp that provides input (possibly indirectly)
    // to node with node_name. A typical indirect chain of nodes (that can occur due
    // to graph inlining) is the following: VarHandleOp -> Identity -> Identity ->
    // ReadVariableOp. Calling the function on any of these nodes would return the
    // name of the VarHandleOp.
    StatusOr<string> GetVarHandleName(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 11 08:05:36 UTC 2023
    - 11.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tfrt/tests/fuse_tpu_compile_and_execute_ops.mlir

      func.return %3 : tensor<*xi32>
    }
    
    }
    
    // -----
    
    module attributes {tf_saved_model.semantics} {
    
    // CHECK-LABEL: func private @reorder_execute_arg_defining_ops
    // CHECK: tf.VarHandleOp
    // CHECK-NEXT: tf.ReadVariableOp
    // CHECK-NEXT: tf.TPUCompileMlirAndExecute
    func.func private @reorder_execute_arg_defining_ops(%arg0: tensor<1x3xf32> {tf.device = "/CPU:0"}) -> (tensor<1x1xf32> {tf.device = "/TPU:0"}) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 13.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables_test.cc

            %0 = "tf.VarHandleOp"() {container = "", device = "/device:CPU:0", shared_name = "var_0"} : () -> tensor<!tf_type.resource<tensor<2xf32>>>
            "tf.AssignVariableOp"(%0, %cst) : (tensor<!tf_type.resource<tensor<2xf32>>>, tensor<2xf32>) -> ()
    
            %cst_0 = "tf.Const"() {device = "", value = dense<[3, 4, 5, 6]> : tensor<4xi32>} : () -> tensor<4xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 16.2K bytes
    - Viewed (0)
  10. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

          "/job:localhost/replica:0/task:0/device:CPU:1"};
      ParallelDevice parallel_device(std::move(devices));
      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op(
          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 08 23:47:35 UTC 2021
    - 15.3K bytes
    - Viewed (0)
Back to top