Lines Matching refs:getNumberOfDimensions

250     uint32_t numberOfDims1 = getNumberOfDimensions(in1);  in calculateBroadcastedShape()
251 uint32_t numberOfDims2 = getNumberOfDimensions(in2); in calculateBroadcastedShape()
330 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthToSpacePrepare()
349 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToDepthPrepare()
370 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
371 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
378 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()
389 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
390 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
391 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 1); in hashtableLookupPrepare()
396 for (uint32_t i = 1; i < getNumberOfDimensions(valueShape); i++) { in hashtableLookupPrepare()
412 uint32_t numInputDims = getNumberOfDimensions(input); in padPrepare()
416 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
439 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in batchToSpacePrepare()
443 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in batchToSpacePrepare()
466 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToBatchPrepare()
470 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in spaceToBatchPrepare()
476 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
505 NN_OPS_CHECK(getNumberOfDimensions(axisShape) == 1); in meanPrepare()
507 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); in meanPrepare()
585 if (getNumberOfDimensions(input) > 1) { in argMinMaxPrepare()
586 output->dimensions.reserve(getNumberOfDimensions(input) - 1); in argMinMaxPrepare()
632 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in groupedConvPrepare()
633 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in groupedConvPrepare()
634 NN_OPS_CHECK(getNumberOfDimensions(bias) == 1); in groupedConvPrepare()