mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-23 19:32:23 +00:00
c10948699b
Rename MakeString to MakeStringWithClassicLocale, MakeStringLite to MakeString, *ParseString to *ParseStringWithClassicLocale. Add missing pass-through versions of MakeStringWithClassicLocale for string types.
29 lines
1 KiB
C++
29 lines
1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "core/framework/data_transfer_manager.h"
|
|
#include "memcpy.h"
|
|
using namespace ONNX_NAMESPACE;
|
|
namespace onnxruntime {
|
|
|
|
Memcpy::Memcpy(const OpKernelInfo& info)
|
|
: OpKernel(info) {
|
|
}
|
|
|
|
Status Memcpy::Compute(OpKernelContext* ctx) const {
|
|
const auto* X = ctx->Input<Tensor>(0);
|
|
Tensor* Y = ctx->Output(0, X->Shape());
|
|
Status retval = Info().GetDataTransferManager().CopyTensor(*X, *Y, Info().GetKernelDef().ExecQueueId());
|
|
|
|
if (!retval.IsOK()) {
|
|
LOGS(ctx->Logger(), ERROR) << MakeString(retval.ErrorMessage(),
|
|
" Copying ", Node().InputDefs()[0]->Name(),
|
|
" to ", Node().OutputDefs()[0]->Name(),
|
|
" Input shape:", X->Shape(), " Output shape:", Y->Shape(),
|
|
" X data:", X->DataRaw(), " Y data:", Y->DataRaw());
|
|
}
|
|
|
|
return retval;
|
|
}
|
|
|
|
} // namespace onnxruntime
|