Lines Matching refs:getNumberOfDimensions

136 uint32_t getNumberOfDimensions(const Shape& shape) {  in getNumberOfDimensions()  function
314 uint32_t numberOfDims1 = getNumberOfDimensions(in1); in calculateBroadcastedShape()
315 uint32_t numberOfDims2 = getNumberOfDimensions(in2); in calculateBroadcastedShape()
365 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthwiseConvPrepare()
366 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in depthwiseConvPrepare()
367 NN_OPS_CHECK(getNumberOfDimensions(bias) == 1); in depthwiseConvPrepare()
399 NN_OPS_CHECK(getNumberOfDimensions(input) <= 4); in genericActivationPrepare()
449 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthToSpacePrepare()
472 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToDepthPrepare()
497 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
498 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
507 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()
521 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
522 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
523 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 1); in hashtableLookupPrepare()
530 for (uint32_t i = 1; i < getNumberOfDimensions(valueShape); i++) { in hashtableLookupPrepare()
548 uint32_t numInputDims = getNumberOfDimensions(input); in padPrepare()
552 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
577 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in batchToSpacePrepare()
581 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in batchToSpacePrepare()
609 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToBatchPrepare()
613 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in spaceToBatchPrepare()
619 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
649 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); in squeezePrepare()
653 NN_OPS_CHECK(getNumberOfDimensions(squeezeDimsShape) == 1); in squeezePrepare()
702 NN_OPS_CHECK(getNumberOfDimensions(axisShape) == 1); in meanPrepare()
704 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); in meanPrepare()
777 uint32_t numInputDims = getNumberOfDimensions(input); in stridedSlicePrepare()
781 NN_OPS_CHECK(getNumberOfDimensions(beginShape) == 1); in stridedSlicePrepare()
782 NN_OPS_CHECK(getNumberOfDimensions(endShape) == 1); in stridedSlicePrepare()
783 NN_OPS_CHECK(getNumberOfDimensions(stridesShape) == 1); in stridedSlicePrepare()
837 output->dimensions.reserve(getNumberOfDimensions(input) - 1); in argMinMaxPrepare()
880 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in groupedConvPrepare()
881 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in groupedConvPrepare()
882 NN_OPS_CHECK(getNumberOfDimensions(bias) == 1); in groupedConvPrepare()