1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
crate::ix!();
#[inline] pub fn workspace_optimizations<T,U>(
nn: *mut NNModule<T,U>,
ws: *mut Workspace,
level: i32) {
todo!();
/*
switch (level) {
case 1:
opt::fuseConvBN(nn, ws);
case 0:
default:
break;
}
*/
}
#[inline] pub fn graph_optimzations<T,U>(nn: *mut NNModule<T,U>, level: i32) {
todo!();
/*
switch (level) {
case 1:
#ifdef USE_NNPACK
opt::addNNPACK(nn, false);
opt::fuseNNPACKConvRelu(nn);
#endif
case 0:
default:
break;
}
*/
}
#[inline] pub fn optimize_with_workspace(
net: NetDef,
ws: *mut Workspace,
level: Option<i32>) -> NetDef
{
let level = level.unwrap_or(1);
todo!();
/*
auto nn = convertToNNModule(net);
graphOptimzations(&nn, level);
workspaceOptimizations(&nn, ws, level);
return convertToCaffe2Proto(nn, net);
*/
}
#[inline] pub fn optimize(
net: NetDef,
level: Option<i32>) -> NetDef
{
let level = level.unwrap_or(1);
todo!();
/*
auto nn = convertToNNModule(net);
graphOptimzations(&nn, level);
return convertToCaffe2Proto(nn, net);
*/
}