Lines Matching refs:useNchw
134 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling() argument
135 InputWithLayout<T_Input> input(useNchw); in roiPooling()
136 OutputWithLayout<T_Input> output(useNchw); in roiPooling()
151 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling() argument
156 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
166 float widthStride, bool useNchw, int8_t* outputData, in roiPooling() argument
171 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
179 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare() local
187 [[maybe_unused]] uint32_t inHeight = getSizeOfDimension(input, useNchw ? 2 : 1); in prepare()
188 [[maybe_unused]] uint32_t inWidth = getSizeOfDimension(input, useNchw ? 3 : 2); in prepare()
189 uint32_t inDepth = getSizeOfDimension(input, useNchw ? 1 : 3); in prepare()
215 if (useNchw) { in prepare()