|
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) |