Searched refs:mUseNchw (Results 1 – 1 of 1) sorted by relevance
160 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout()165 if (mUseNchw) { in initialize()171 const T* getNhwcBuffer() { return mUseNchw ? mDataNhwc.data() : mDataOriginal; } in getNhwcBuffer()178 bool mUseNchw; variable184 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout()190 if (mUseNchw) { in initialize()198 T* getNhwcBuffer() { return mUseNchw ? mDataNhwc.data() : mDataOriginal; } in getNhwcBuffer()201 if (mUseNchw) { in commit()211 bool mUseNchw; variable