Searched refs:mUseNchw (Results 1 – 1 of 1) sorted by relevance
159 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout()164 if (mUseNchw) { in initialize()170 const T* getNhwcBuffer() { return mUseNchw ? mDataNhwc.data() : mDataOriginal; } in getNhwcBuffer()177 bool mUseNchw; variable183 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout()189 if (mUseNchw) { in initialize()197 T* getNhwcBuffer() { return mUseNchw ? mDataNhwc.data() : mDataOriginal; } in getNhwcBuffer()200 if (mUseNchw) { in commit()210 bool mUseNchw; variable