[dnn/batchnorm]: added some more code in Triton-C batchnorm implementations
This commit is contained in:
@@ -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));
|
||||
}
|
||||
|
Reference in New Issue
Block a user