saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-17 21:10:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cherry pick fix constant pow (#18785)

Browse source 
### Description
Cherry pick https://github.com/microsoft/onnxruntime/pull/18784
This commit is contained in:
Jake Mathern 2023-12-11 17:41:16 -08:00 committed by Jeff Bloomfield
parent 107d7492b9
commit d2f7a5b128
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorElementWise.cpp
View file

@ -479,7 +479,7 @@ public:
ML_CHECK_VALID_ARGUMENT(kernelInfo.GetInputCount() == 2);
ML_CHECK_VALID_ARGUMENT(kernelInfo.GetOutputCount() == 1);
auto constExpTensor = kernelInfo.TryGetConstantInputTensor(1);
auto constExpTensor = kernelInfo.TryGetConstantCpuInputTensor(1);
if (constExpTensor && constExpTensor->GetTotalElementCount() == 1)
{
std::vector<std::optional<uint32_t>> kernelInputIndices = {0};

4
onnxruntime/core/providers/dml/OperatorAuthorHelper/MLOperatorAuthorHelper.h
View file

@ -605,11 +605,11 @@ public:
return MLOperatorTensor(tensor.Get());
}
std::optional<MLOperatorTensor> TryGetConstantInputTensor(uint32_t inputIndex) const
std::optional<MLOperatorTensor> TryGetConstantCpuInputTensor(uint32_t inputIndex) const
{
Microsoft::WRL::ComPtr<IMLOperatorTensor> tensor;
ORT_THROW_IF_FAILED(m_implPrivate->TryGetConstantInputTensor(inputIndex, &tensor));
if (tensor)
if (tensor && tensor->IsCpuData())
{
return MLOperatorTensor(tensor.Get());
}