|
ONE - On-device Neural Engine
|
#include "ir/train/operation/BinaryArithmetic.h"#include "ir/train/operation/Conv2D.h"#include "ir/train/operation/DepthwiseConv2D.h"#include "ir/train/operation/ElementwiseActivation.h"#include "ir/train/operation/FullyConnected.h"#include "ir/train/operation/Loss.h"#include "ir/train/operation/Pad.h"#include "ir/train/operation/Permute.h"#include "ir/train/operation/Pool2D.h"#include "ir/train/operation/Reduce.h"#include "ir/train/operation/Reshape.h"#include "ir/train/operation/Softmax.h"