saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-18 18:52:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add in a couple more missing GetType methods

Browse source
This commit is contained in:
Ryan Hill 2021-05-12 17:17:32 -07:00
parent b1924a9323
commit 1a20b6d13e
3 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/framework/provider_bridge_ort.cc
View file

@ -492,6 +492,8 @@ struct ProviderHostImpl : ProviderHost {
MLDataType DataTypeImpl__GetType_uint64() override { return DataTypeImpl::GetType<uint64_t>(); }
MLDataType DataTypeImpl__GetType_float() override { return DataTypeImpl::GetType<float>(); }
MLDataType DataTypeImpl__GetType_double() override { return DataTypeImpl::GetType<double>(); }
MLDataType DataTypeImpl__GetType_BFloat16() override { return DataTypeImpl::GetType<BFloat16>(); }
MLDataType DataTypeImpl__GetType_MLFloat16() override { return DataTypeImpl::GetType<MLFloat16>(); }
MLDataType DataTypeImpl__GetTensorType_bool() override { return DataTypeImpl::GetTensorType<bool>(); }
MLDataType DataTypeImpl__GetTensorType_int8() override { return DataTypeImpl::GetTensorType<int8_t>(); }
MLDataType DataTypeImpl__GetTensorType_uint8() override { return DataTypeImpl::GetTensorType<uint8_t>(); }

4
onnxruntime/core/providers/shared_library/provider_bridge_provider.cc
View file

@ -91,6 +91,10 @@ MLDataType DataTypeImpl::GetType<float>() { return Provider_GetHost()->DataTypeI
template <>
MLDataType DataTypeImpl::GetType<double>() { return Provider_GetHost()->DataTypeImpl__GetType_double(); }
template <>
MLDataType DataTypeImpl::GetType<BFloat16>() { return Provider_GetHost()->DataTypeImpl__GetType_BFloat16(); }
template <>
MLDataType DataTypeImpl::GetType<MLFloat16>() { return Provider_GetHost()->DataTypeImpl__GetType_MLFloat16(); }
template <>
MLDataType DataTypeImpl::GetTensorType<bool>() { return Provider_GetHost()->DataTypeImpl__GetTensorType_bool(); }
template <>
MLDataType DataTypeImpl::GetTensorType<int8_t>() { return Provider_GetHost()->DataTypeImpl__GetTensorType_int8(); }

2
onnxruntime/core/providers/shared_library/provider_interfaces.h
View file

@ -413,6 +413,8 @@ struct ProviderHost {
virtual MLDataType DataTypeImpl__GetType_uint64() = 0;
virtual MLDataType DataTypeImpl__GetType_float() = 0;
virtual MLDataType DataTypeImpl__GetType_double() = 0;
virtual MLDataType DataTypeImpl__GetType_BFloat16() = 0;
virtual MLDataType DataTypeImpl__GetType_MLFloat16() = 0;
virtual MLDataType DataTypeImpl__GetTensorType_bool() = 0;
virtual MLDataType DataTypeImpl__GetTensorType_int8() = 0;
virtual MLDataType DataTypeImpl__GetTensorType_uint8() = 0;