[dnn/batchnorm]: added some more code in Triton-C batchnorm implementations

This commit is contained in:
Philippe Tillet
2019-07-08 20:18:20 -07:00
parent fa3270dcf2
commit f74dcb7e30
13 changed files with 103 additions and 11 deletions

View File

@@ -320,6 +320,10 @@ value *builder::create_trans(value *A, const std::string &name) {
return insert(trans_inst::create(A, name));
}
value *builder::create_sqrt(value *A, const std::string &name) {
return insert(sqrt_inst::create(A, name));
}
value *builder::create_reduce(value *A, const std::string &name) {
return insert(reduce_inst::create(A, name));
}