[dnn] Adding batchnorm

This commit is contained in:
Philippe Tillet
2019-07-08 18:44:37 -07:00
parent b0cf3143c5
commit f9db0449b7
42 changed files with 682 additions and 1763 deletions

View File

@@ -566,6 +566,19 @@ instruction* trans_inst::create(value *arg, const std::string &name, instruction
return new trans_inst(arg, name, next);
}
//===----------------------------------------------------------------------===//
// reduce instructions
//===----------------------------------------------------------------------===//
reduce_inst::reduce_inst(value *arg, const std::string &name, instruction *next)
: builtin_inst(arg->get_type()->get_scalar_ty(), 1, 1, name, next) {
set_operand(0, arg);
}
instruction* reduce_inst::create(value *arg, const std::string &name, instruction *next) {
return new reduce_inst(arg, name, next);
}
//===----------------------------------------------------------------------===//
// select instructions
//===----------------------------------------------------------------------===//