Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for VarHandleOp (0.2 sec)

  1. 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);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  2. 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);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
Back to top