ONE - On-device Neural Engine
|
#include "nnop/PadInfo.h"
#include "nnop/StrideInfo.h"
#include <nncc/core/ADT/feature/Shape.h>
#include <nncc/core/ADT/feature/Reader.h>
#include <nncc/core/ADT/feature/Accessor.h>
#include <nncc/core/ADT/kernel/Shape.h>
#include <nncc/core/ADT/kernel/Reader.h>
Go to the source code of this file.
Namespaces | |
namespace | nnop |
Functions | |
template<typename OutputDType , typename InputDType , typename KernelDType > | |
void | nnop::conv (const nncc::core::ADT::feature::Shape &out_shape, nncc::core::ADT::feature::Accessor< OutputDType > &out_data, const nncc::core::ADT::feature::Shape &in_shape, const nncc::core::ADT::feature::Reader< InputDType > &in_data, const nncc::core::ADT::kernel::Shape &ker_shape, const nncc::core::ADT::kernel::Reader< KernelDType > &ker_data, const PadInfo &pad_info, const StrideInfo &stride_info) |