Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for attribute (0.19 sec)

  1. tensorflow/c/c_api_test.cc

      TF_DeleteTensor(fetchValues[0]);
      TF_DeleteTensor(fetchValues[1]);
    }
    
    // REGISTER_OP for CApiAttributesTest test cases.
    // Registers two ops, each with a single attribute called 'v'.
    // The attribute in one op will have a type 'type', the other
    // will have list(type).
    #define ATTR_TEST_REGISTER_OP(type)                           \
      REGISTER_OP("CApiAttributesTestOp" #type)                   \
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  2. tensorflow/c/eager/c_api_test.cc

      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);
      unsigned char is_list = 0;
      ASSERT_EQ(TF_ATTR_TYPE,
                TFE_OpGetAttrType(copy_op, "dtype", &is_list, status));
    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)
Back to top