• Home
  • Raw
  • Download

Lines Matching refs:getNumberOfDimensions

154 uint32_t getNumberOfDimensions(const Shape& shape) {  in getNumberOfDimensions()  function
341 uint32_t numberOfDims1 = getNumberOfDimensions(in1); in calculateBroadcastedShape()
342 uint32_t numberOfDims2 = getNumberOfDimensions(in2); in calculateBroadcastedShape()
421 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthToSpacePrepare()
440 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToDepthPrepare()
461 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
462 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
469 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()
480 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
481 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
482 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 1); in hashtableLookupPrepare()
487 for (uint32_t i = 1; i < getNumberOfDimensions(valueShape); i++) { in hashtableLookupPrepare()
503 uint32_t numInputDims = getNumberOfDimensions(input); in padPrepare()
507 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
530 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in batchToSpacePrepare()
534 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in batchToSpacePrepare()
557 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToBatchPrepare()
561 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in spaceToBatchPrepare()
567 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
596 NN_OPS_CHECK(getNumberOfDimensions(axisShape) == 1); in meanPrepare()
598 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); in meanPrepare()
676 if (getNumberOfDimensions(input) > 1) { in argMinMaxPrepare()
677 output->dimensions.reserve(getNumberOfDimensions(input) - 1); in argMinMaxPrepare()
723 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in groupedConvPrepare()
724 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in groupedConvPrepare()
725 NN_OPS_CHECK(getNumberOfDimensions(bias) == 1); in groupedConvPrepare()