Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_DataTypeSize (0.08 sec)

  1. tensorflow/c/c_api_test.cc

      EXPECT_EQ(TF_HALF, static_cast<TF_DataType>(tensorflow::DT_HALF));
      EXPECT_EQ(TF_DataTypeSize(TF_DOUBLE),
                tensorflow::DataTypeSize(tensorflow::DT_DOUBLE));
      EXPECT_EQ(TF_DataTypeSize(TF_STRING),
                tensorflow::DataTypeSize(tensorflow::DT_STRING));
      // Test with invalid type; should always return 0 as documented
      EXPECT_EQ(TF_DataTypeSize(static_cast<TF_DataType>(0)), 0);
    }
    
    TEST(CAPI, StatusEnum) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  2. tensorflow/c/eager/dlpack.cc

      delete owner;
    }
    
    // Converts TF_DATAType to DLPack data type.
    DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) {
      DLDataType dtype;
      dtype.lanes = 1;
      dtype.bits = TF_DataTypeSize(data_type) * 8;
      switch (data_type) {
        case TF_DataType::TF_BOOL:
          dtype.code = DLDataTypeCode::kDLBool;
          break;
        case TF_DataType::TF_HALF:
        case TF_DataType::TF_FLOAT:
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
Back to top