Lines Matching refs:getSizeOfDimension
158 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension() function
348 dim1 = getSizeOfDimension(in1, numberOfDims1 - i); in calculateBroadcastedShape()
352 dim2 = getSizeOfDimension(in2, numberOfDims2 - i); in calculateBroadcastedShape()
424 uint32_t batches = getSizeOfDimension(input, 0); in depthToSpacePrepare()
425 uint32_t height = getSizeOfDimension(input, 1); in depthToSpacePrepare()
426 uint32_t width = getSizeOfDimension(input, 2); in depthToSpacePrepare()
427 uint32_t channels = getSizeOfDimension(input, 3); in depthToSpacePrepare()
443 uint32_t batches = getSizeOfDimension(input, 0); in spaceToDepthPrepare()
444 uint32_t height = getSizeOfDimension(input, 1); in spaceToDepthPrepare()
445 uint32_t width = getSizeOfDimension(input, 2); in spaceToDepthPrepare()
446 uint32_t channels = getSizeOfDimension(input, 3); in spaceToDepthPrepare()
464 const uint32_t columns = getSizeOfDimension(valueShape, 1); in embeddingLookupPrepare()
465 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); in embeddingLookupPrepare()
470 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in embeddingLookupPrepare()
484 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); in hashtableLookupPrepare()
488 outputShape->dimensions.push_back(getSizeOfDimension(valueShape, i)); in hashtableLookupPrepare()
508 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
509 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in padPrepare()
517 outDims[i] = beforePadding + getSizeOfDimension(input, i) + afterPadding; in padPrepare()
536 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in batchToSpacePrepare()
538 uint32_t batches = getSizeOfDimension(input, 0); in batchToSpacePrepare()
539 uint32_t height = getSizeOfDimension(input, 1); in batchToSpacePrepare()
540 uint32_t width = getSizeOfDimension(input, 2); in batchToSpacePrepare()
541 uint32_t channels = getSizeOfDimension(input, 3); in batchToSpacePrepare()
563 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in spaceToBatchPrepare()
568 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == 2); in spaceToBatchPrepare()
569 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in spaceToBatchPrepare()
571 uint32_t batches = getSizeOfDimension(input, 0); in spaceToBatchPrepare()
572 uint32_t height = getSizeOfDimension(input, 1); in spaceToBatchPrepare()
573 uint32_t width = getSizeOfDimension(input, 2); in spaceToBatchPrepare()
574 uint32_t channels = getSizeOfDimension(input, 3); in spaceToBatchPrepare()
599 int32_t axisSize = static_cast<int32_t>(getSizeOfDimension(axisShape, 0)); in meanPrepare()
615 outDims[idx] = getSizeOfDimension(input, idx); in meanPrepare()
652 outDims[idx - numSkipAxis] = getSizeOfDimension(input, idx); in meanPrepare()
727 NN_OPS_CHECK(getSizeOfDimension(filter, 0) == getSizeOfDimension(bias, 0)); in groupedConvPrepare()
729 NN_OPS_CHECK(getSizeOfDimension(filter, 3) * numGroups == getSizeOfDimension(input, 3)); in groupedConvPrepare()
730 NN_OPS_CHECK(getSizeOfDimension(filter, 0) % numGroups == 0); in groupedConvPrepare()
732 uint32_t channels_out = getSizeOfDimension(filter, 0); in groupedConvPrepare()
733 uint32_t width = getSizeOfDimension(input, 2); in groupedConvPrepare()
734 uint32_t height = getSizeOfDimension(input, 1); in groupedConvPrepare()
735 uint32_t filterWidth = getSizeOfDimension(filter, 2); in groupedConvPrepare()
736 uint32_t filterHeight = getSizeOfDimension(filter, 1); in groupedConvPrepare()
737 uint32_t batches = getSizeOfDimension(input, 0); in groupedConvPrepare()