#include "internal/layers/ArgMinMax.h"
#include <arm_compute/runtime/CL/CLScheduler.h>
-#include <arm_compute/core/CL/CLHelpers.h>
void ArgMinMax::configure(::arm_compute::ITensor *input, ::arm_compute::ITensor *output,
std::vector<uint32_t> axis, int rank, bool is_min, bool is_max)
#define __ARG_MIN_MAX_H__
#include "internal/arm_compute.h"
-#include "internal/arm_compute/Cast.h"
-#include <arm_compute/core/ITensor.h>
-#include <arm_compute/runtime/IFunction.h>
class ArgMinMax : public ::arm_compute::IFunction
{
#include "GenericFullyConnectedLayer.h"
#include "internal/arm_compute.h"
-#include <arm_compute/core/Helpers.h>
-
void GenericFullyConnectedLayer::configure(::arm_compute::ITensor *input,
::arm_compute::ITensor *weights,
::arm_compute::ITensor *biases,
#ifndef __GENERIC_FULLY_CONNECTED_LAYER_H__
#define __GENERIC_FULLY_CONNECTED_LAYER_H__
-#include <arm_compute/runtime/Tensor.h>
-#include <arm_compute/runtime/CL/CLTensor.h>
#include <arm_compute/runtime/CL/functions/CLFullyConnectedLayer.h>
#include <arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h>
#include "internal/layers/GenericReshapeLayer.h"
#include "internal/layers/HashtableLookupLayer.h"
#include <arm_compute/runtime/CL/CLScheduler.h>
-#include <map>
-#include <cstring>
void HashtableLookupLayer::configure(::arm_compute::ITensor *lookups, ::arm_compute::ITensor *keys,
::arm_compute::ITensor *values, ::arm_compute::ITensor *output,
#define __HASHTABLE_LOOKUP_H__
#include "internal/arm_compute.h"
-#include <arm_compute/core/ITensor.h>
-#include <arm_compute/runtime/IFunction.h>
-#include <vector>
class HashtableLookupLayer : public ::arm_compute::IFunction
{
#include "internal/arm_compute.h"
#include "internal/arm_compute/Cast.h"
-#include <arm_compute/core/ITensor.h>
-#include <arm_compute/runtime/IFunction.h>
class SimpleBatchToSpaceND : public ::arm_compute::IFunction
{
#ifndef __SIMPLE_CAST_LAYER_H__
#define __SIMPLE_CAST_LAYER_H__
-#include <arm_compute/core/ITensor.h>
-
#include "internal/arm_compute.h"
#include "internal/arm_compute/Cast.h"
#include "internal/arm_compute.h"
#include "internal/arm_compute/Cast.h"
-#include <arm_compute/core/ITensor.h>
-#include <arm_compute/runtime/IFunction.h>
class SimpleDepthToSpace : public ::arm_compute::IFunction
{
#define __SIMPLE_EMBEDDING_LOOKUP_H__
#include "internal/arm_compute.h"
-#include <arm_compute/core/ITensor.h>
-#include <arm_compute/runtime/IFunction.h>
/**
* @file SimpleEmbeddingLookup.h
#define __SIMPLE_NEG_H__
#include "internal/arm_compute.h"
-#include <arm_compute/core/ITensor.h>
-#include <arm_compute/runtime/IFunction.h>
class SimpleNeg : public ::arm_compute::IFunction
{