本文整理汇总了C++中BlockInfo::FMSize方法的典型用法代码示例。如果您正苦于以下问题:C++ BlockInfo::FMSize方法的具体用法?C++ BlockInfo::FMSize怎么用?C++ BlockInfo::FMSize使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类BlockInfo
的用法示例。
在下文中一共展示了BlockInfo::FMSize方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。
示例1:
void MKLPoolingLayer<Dtype>::pack_buffer(FeatureMap *fm, Dtype *to, const Dtype *from) {
for (int i = 0; i < fm->NumPackBlocks(); i++) {
BlockInfo * bi = fm->GetPackBlock(i);
int bMBLen = bi->MBLen();
int bMBStart = bi->MBStart();
int bFMLen = bi->FMLen();
int bFMStart = bi->FMStart();
Dtype *src = (Dtype*) from;
Dtype *dst = (Dtype*) (to + bi->BufOffset());
for (int mb = 0; mb < bMBLen; mb++) {
for (int fm = 0; fm < bFMLen; fm++) {
for (int s = 0 ; s < bi->FMSize(); s++) {
//dst[fm][mb][s] = src[s][bFMStart+fm][bMBStart+mb]; //[bMBStart+mb][bFMStart+fm][s];
dst[(fm*bMBLen + mb)*bi->FMSize() + s] =
src[s*bFMLen*bMBLen + (bFMStart+fm)*bMBLen + (bMBStart+mb)];
}
}
}
}
}