Search Options

Results per page
Sort
Preferred Languages
Advance

Results 91 - 100 of 114 for var_handle_op (0.55 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_initializer_function_ops_to_main.mlir

          %out_0, %ctl_0 = tf_executor.island wraps "tf.Const"() {value = dense<"var_0"> : tensor<1x!tf_type.string>} : () -> tensor<1x!tf_type.string>
          %out_1, %ctl_1 = tf_executor.island wraps "tf.VarHandleOp"() {container = "", shared_name = "var_0", device = "/device:CPU:0"} : () -> tensor<!tf_type.resource<tensor<2xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:49:35 UTC 2023
    - 29.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/device_canonicalize.mlir

    func.func @cluster_result_for_resource_update(%arg0 : tensor<!tf_type.string>, %arg1 : tensor<!tf_type.string>) -> (tensor<!tf_type.string>, tensor<!tf_type.string>) {
      %resource = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf_type.resource<tensor<*x!tf_type.string>>>
    
      // CHECK: %[[RESULT:.*]] = "tf_device.cluster"
      %0:2 = "tf_device.cluster"() ({
        // CHECK: "tf.Equal"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Nov 04 14:07:37 UTC 2022
    - 6.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tf_savedmodel_passes.td

        for those which come from a "bound input" parameter. For the latter,
        said parameter will not be removed, unlike LowerGlobalsToMlProgramPass.
    
        So this will lower e.g.
    
          %0 = "tf.VarHandleOp" {name = "foo"}
          %1 = "tf.ReadVariableOp"(%0)
    
        to
    
          %0 = ml_program.global_load @vars.foo
      }];
    
      let constructor = "::mlir::tf_saved_model::CreateLowerVariableOpsToMlProgramPass()";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 09 19:11:34 UTC 2023
    - 7.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc

          if (auto global = tf_saved_model::LookupBoundInputOfType<
                  tf_saved_model::GlobalTensorOp>(func, i, syms)) {
            OpBuilder builder(func.getBody());
            auto dummy = builder.create<TF::VarHandleOp>(
                global.getLoc(), func.getArgument(i).getType(), "dummy", "dummy");
            func.getArgument(i).replaceAllUsesWith(dummy.getResult());
            argsToErase.set(i);
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_test.cc

      TFE_DeleteContextOptions(opts);
    
      TFE_Op* var_op = TFE_NewOp(ctx, "VarHandleOp", status);
      TFE_OpSetAttrType(var_op, "dtype", TF_INT64);
      TFE_OpSetAttrShape(var_op, "shape", {}, 0, status);
      const TFE_OpAttrs* attributes = TFE_OpGetAttrs(var_op);
    
      TFE_Op* copy_op = TFE_NewOp(ctx, "VarHandleOp", status);
      TFE_OpSetAttrType(copy_op, "dtype", TF_FLOAT);
      TFE_OpAddAttrs(copy_op, attributes);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_test_util.cc

    TFE_TensorHandle* TestVariable(TFE_Context* ctx, float value,
                                   const tensorflow::string& device_name) {
      TF_Status* status = TF_NewStatus();
      // Create the variable handle.
      TFE_Op* op = TFE_NewOp(ctx, "VarHandleOp", status);
      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op, "shape", {}, 0, status);
      TFE_OpSetAttrString(op, "container", "localhost", 0);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/tests/convert_control_to_data_outputs.mlir

    // Tests loop with two resource types, one of them being unique per iteration.
    //
    // Similar to above test but with one additional resource that is not unique per
    // iteration (created by `tf.VarHandleOp`).
    
    func.func @mixed_unique_resource_chain(%arg0: tensor<i32>, %arg1: tensor<f32>) {
      tf_executor.graph {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 18:35:00 UTC 2024
    - 68.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/tests/mark_ops_for_outside_compilation.mlir

      "tf_device.cluster"() ({
        // CHECK:      "tf.VarHandleOp"
        // CHECK-SAME: _xla_outside_compilation
        "tf.VarHandleOp"() {allowed_devices = [], container = "", device = "", shared_name = ""} : () -> tensor<!tf_type.resource<tensor<!tf_type.string>>>
        tf_device.return
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 16:22:32 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/resource_device_inference.cc

      // these regions, so use a pre-order walk.
      WalkResult walk_res = func_op.walk<WalkOrder::PreOrder>([&](Operation* op) {
        if (auto var_handle = dyn_cast<VarHandleOp>(op)) {
          // Record VarHandleOp's device attribute.
          StringRef device_attr = GetDeviceAttr(op);
          if (device_attr.empty()) return WalkResult::advance();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 03 03:47:00 UTC 2023
    - 13.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tfrt/tests/analysis/cost_analysis.mlir

    }
    
    // CHECK-LABEL: test_expensive_ops
    func.func @test_expensive_ops(%arg: tensor<?x512xf32>) -> tensor<?x512xf32> {
        // expected-remark@+1 {{Cost: 1}}
        %0 = "tf.VarHandleOp"() {allowed_devices = [], container = "", device = "/job:localhost/replica:0/task:0/device:CPU:0", shared_name = "var"} : () -> tensor<!tf_type.resource<tensor<512x512xf32>>>
        // expected-remark@+1 {{Cost: 2}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 14 15:35:49 UTC 2023
    - 12.2K bytes
    - Viewed (0)
Back to top