Lines Matching refs:outDim
3230 uint32_t outDim[] = {4, 2, 2, 1}; in roiAlignOpTest() local
3239 {getOpType(inputOperandCode, 4, outDim)}); in roiAlignOpTest()
3265 uint32_t outDim[] = {4, 2, 2, 1}; in roiPoolingOpTest() local
3273 {getOpType(inputOperandCode, 4, outDim)}); in roiPoolingOpTest()
3353 uint32_t outDim[] = {1, 2, 2, 2}; in groupedConvOpTest() local
3379 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outDim); in groupedConvOpTest()
3428 uint32_t outDim[] = {1, 5, 5, 2}, outShapeDim[] = {4}; in transposeConvOpTest() local
3452 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outDim); in transposeConvOpTest()
3660 uint32_t outDim[] = {5, 8}; in axisAlignedBBoxTransformOpTest() local
3666 {getOpType(roiOperandCode, 2, outDim)}); in axisAlignedBBoxTransformOpTest()
4736 for (uint32_t inDim = 0, outDim = 0; outDim < outputRank; ++outDim) { in packTest() local
4737 if (outDim == axis) { in packTest()
4738 outputDimensions[outDim] = inputTensorCount; in packTest()
4740 outputDimensions[outDim] = inputDimensions[inDim++]; in packTest()