#include <arm_compute/core/TensorShape.h>
#include <arm_compute/core/TensorInfo.h>
#include "../IO_accessor.h"
+#include "../shape.h"
#include <cassert>
-namespace android {
-namespace nn {
-
-// TODO remove from this source and use it from runtime
-uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);
-
-} // namespace nn
-} // namespace android
-
namespace nnfw {
namespace kernel {
namespace acl {
-arm_compute::TensorShape fromNNShape(const android::nn::Shape& shape);
bool maxPoolFloat32(const float* inputData, const android::nn::Shape& inputShape,
int32_t padding_left, int32_t padding_right,
int32_t activation,
float* outputData, const android::nn::Shape& outputShape)
{
- arm_compute::TensorShape input_shape = fromNNShape(inputShape);
- arm_compute::TensorShape output_shape = fromNNShape(outputShape);
+ arm_compute::TensorShape input_shape = util::fromNNShape(inputShape);
+ arm_compute::TensorShape output_shape = util::fromNNShape(outputShape);
assert(activation == ANEURALNETWORKS_FUSED_NONE || activation == ANEURALNETWORKS_FUSED_RELU);