Searched refs:mUseNchw (Results 1 – 1 of 1) sorted by relevance
141 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout()146 if (mUseNchw) { in initialize()152 const T* getNhwcBuffer() { return mUseNchw ? mDataNhwc.data() : mDataOriginal; } in getNhwcBuffer()159 bool mUseNchw; variable165 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout()171 if (mUseNchw) { in initialize()179 T* getNhwcBuffer() { return mUseNchw ? mDataNhwc.data() : mDataOriginal; } in getNhwcBuffer()182 if (mUseNchw) { in commit()192 bool mUseNchw; variable