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

Fix nuphar 2

Browse source
This commit is contained in:
Ryan Hill 2021-05-10 13:54:34 -07:00
parent 7cc57c11c8
commit cd2c891a34
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
onnxruntime/core/providers/cpu/nn/pool_attributes.h
View file

@ -6,7 +6,6 @@
#include <cmath>
#ifndef SHARED_PROVIDER
#include "core/common/common.h"
#include "core/framework/op_kernel.h"
#include "core/framework/op_node_proto_helper.h"
#include "core/framework/tensor_shape.h"
#include "core/providers/common.h"
@ -20,7 +19,13 @@ struct PoolAttributes {
return op_name == "GlobalAveragePool" || op_name == "GlobalMaxPool" || op_name == "GlobalLpPool";
}
#ifdef SHARED_PROVIDER
// Shared providers don't know about OpNodeProtoHelper
PoolAttributes(const OpKernelInfo& info,
#else
// Providers like Nuphar don't know about OpKernelInfo
PoolAttributes(const OpNodeProtoHelper<ProtoHelperNodeContext>& info,
#endif
const std::string& op_name, int start_version)
: global_pooling(IsGlobalPooling(op_name)) {
if (global_pooling) {