提交 a5a29826 编写于 作者: M Megvii Engine Team

feat(mgb/dnn): add accuracy_depend_on_batch attribute

GitOrigin-RevId: 2be1db43d06b3d9712a8faeb489f67182a9ed16c
上级 4b141f8d
......@@ -127,6 +127,15 @@ public:
* \brief whether the algo is usable once shape changed.
* */
USABLE_DEPEND_ON_SHAPE = 1 << 2,
/**
* \brief whether the accuracy of the algo is dependent with respect
* to batch
* In the case of using algorithm with this attribute, even if the
* content of each batch is the same, the output under multiple batch
* input and single batch input may not equal
* */
ACCURACY_DEPEND_ON_BATCH = 1 << 3,
};
/**
......
......@@ -19,7 +19,8 @@ using namespace megdnn;
cb(DEFAULT) \
cb(REPRODUCIBLE) \
cb(NAIVE) \
cb(USABLE_DEPEND_ON_SHAPE)
cb(USABLE_DEPEND_ON_SHAPE) \
cb(ACCURACY_DEPEND_ON_BATCH)
namespace {
inline const char* attr_str(const AlgoAttribute& attr) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册