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

PR feedback

Browse source
This commit is contained in:
Dwayne Robinson 2020-03-27 19:32:47 -07:00
parent f1a062c292
commit 5972ce5566
2 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
onnxruntime/core/providers/dml/DmlExecutionProvider/src/ExecutionProvider.cpp
View file

@ -500,11 +500,12 @@ namespace Dml
std::string partitionKernelPrefix = std::to_string(m_partitionKernelPrefixVal++) + "_";
uint32_t deviceDataTypeMask = GetSuppportedDeviceDataTypeMask();
return PartitionGraph(graph,
return PartitionGraph(
graph,
*m_internalRegInfoMap,
registries,
deviceDataTypeMask,
m_kernelRegistry.get(),
registries,
deviceDataTypeMask,
m_kernelRegistry.get(),
partitionKernelPrefix
);
}

4
onnxruntime/core/providers/dml/OperatorAuthorHelper/OperatorHelper.cpp
View file

@ -155,14 +155,14 @@ namespace OperatorHelper
memcpy(data, tensor.GetByteData(), elementByteSize);
}
int64_t ReadScalarTensorAsInt64(const MLOperatorTensor& tensor)
int64_t ReadScalarTensorCastToInt64(const MLOperatorTensor& tensor)
{
std::byte tensorBytes[8];
ReadScalarTensorData(tensor, /*out*/ &tensorBytes, sizeof(tensorBytes));
return CastToInt64(tensor.GetTensorDataType(), &tensorBytes);
}
double ReadScalarTensorAsFloat64(const MLOperatorTensor& tensor)
double ReadScalarTensorCastToFloat64(const MLOperatorTensor& tensor)
{
std::byte tensorBytes[8];
ReadScalarTensorData(tensor, /*out*/ &tensorBytes, sizeof(tensorBytes));