mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-26 22:35:43 +00:00
f92e47e95b
There's a circular dependency between onnxruntime_util and onnxruntime_framework. Remove onnxruntime_util's dependency on onnxruntime_framework.
41 lines
1.1 KiB
C++
41 lines
1.1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "contrib_ops/cpu/element_wise_ops.h"
|
|
|
|
#include "core/framework/math.h"
|
|
#include "core/providers/cpu/math/element_wise_ops.h"
|
|
|
|
namespace onnxruntime {
|
|
namespace contrib {
|
|
|
|
ONNX_CPU_OPERATOR_KERNEL(
|
|
Affine,
|
|
1,
|
|
KernelDefBuilder().TypeConstraint("T", DataTypeImpl::GetTensorType<float>()),
|
|
Affine<float>);
|
|
|
|
template <>
|
|
Status Affine<float>::Compute(OpKernelContext* ctx) const {
|
|
auto& X = *ctx->Input<Tensor>(0);
|
|
auto& Y = *ctx->Output(0, X.Shape());
|
|
MakeEigenArrayMap<float>(Y) = alpha_ * MakeEigenArrayMap<float>(X) + beta_;
|
|
return Status::OK();
|
|
}
|
|
|
|
ONNX_CPU_OPERATOR_KERNEL(
|
|
Scale,
|
|
1,
|
|
KernelDefBuilder().TypeConstraint("T", DataTypeImpl::GetTensorType<float>()),
|
|
Scale<float>);
|
|
|
|
template <>
|
|
Status Scale<float>::Compute(OpKernelContext* ctx) const {
|
|
auto& X = *ctx->Input<Tensor>(0);
|
|
auto& Y = *ctx->Output(0, X.Shape());
|
|
EigenMap<float>(Y) = scale_ * EigenMap<float>(X);
|
|
return Status::OK();
|
|
}
|
|
|
|
} // namespace contrib
|
|
} // namespace onnxruntime
|