Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_DataTypeSize (0.2 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) {
    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/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:
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
Back to top