Home
last modified time | relevance | path

Searched refs:getNumberOfDimensions (Results 1 – 25 of 65) sorted by relevance

123

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp91 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
100 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2u); in prepare()
105 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2u); in prepare()
111 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2u); in prepare()
117 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2u); in prepare()
121 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2u); in prepare()
127 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2u); in prepare()
133 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2u); in prepare()
137 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2u); in prepare()
151 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1u); in prepare()
[all …]
DQLSTM.cpp64 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
71 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2u); in prepare()
76 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2u); in prepare()
82 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2u); in prepare()
88 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2u); in prepare()
92 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2u); in prepare()
98 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2u); in prepare()
104 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2u); in prepare()
108 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2u); in prepare()
122 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1u); in prepare()
[all …]
DBidirectionalSequenceRNN.cpp334 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3u); in prepare()
335 NN_RET_CHECK_EQ(getNumberOfDimensions(fwWeights), 2u); in prepare()
336 NN_RET_CHECK_EQ(getNumberOfDimensions(fwRecurrentWeights), 2u); in prepare()
337 NN_RET_CHECK_EQ(getNumberOfDimensions(fwBias), 1u); in prepare()
338 NN_RET_CHECK_EQ(getNumberOfDimensions(fwHiddenState), 2u); in prepare()
339 NN_RET_CHECK_EQ(getNumberOfDimensions(bwWeights), 2u); in prepare()
340 NN_RET_CHECK_EQ(getNumberOfDimensions(bwRecurrentWeights), 2u); in prepare()
341 NN_RET_CHECK_EQ(getNumberOfDimensions(bwBias), 1u); in prepare()
342 NN_RET_CHECK_EQ(getNumberOfDimensions(bwHiddenState), 2u); in prepare()
361 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3u); in prepare()
[all …]
DBatchMatmul.cpp65 uint32_t numDims = getNumberOfDimensions(LHSTensorShape); in computeOutputDimensions()
77 uint32_t numDims = getNumberOfDimensions(shape); in swapRowColumnDims()
88 int rank = getNumberOfDimensions(inputShape); in transposeRowsColumns()
206 NN_RET_CHECK_EQ(getNumberOfDimensions(inputLHSTensorShape), in prepare()
207 getNumberOfDimensions(inputRHSTensorShape)) in prepare()
209 NN_RET_CHECK_GE(getNumberOfDimensions(inputLHSTensorShape), 2u) in prepare()
211 NN_RET_CHECK_LE(getNumberOfDimensions(inputLHSTensorShape), 4u) in prepare()
213 uint32_t numDims = getNumberOfDimensions(inputLHSTensorShape); in prepare()
DStridedSlice.cpp59 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in compute()
100 uint32_t numInputDims = getNumberOfDimensions(inputShape); in prepare()
107 NN_OPS_CHECK(getNumberOfDimensions(beginShape) == 1); in prepare()
108 NN_OPS_CHECK(getNumberOfDimensions(endShape) == 1); in prepare()
109 NN_OPS_CHECK(getNumberOfDimensions(stridesShape) == 1); in prepare()
DSlice.cpp51 std::vector<uint32_t> outputIndex(getNumberOfDimensions(outputShape), 0); in evalGeneric()
53 std::vector<uint32_t> inputIndex(getNumberOfDimensions(inputShape)); in evalGeneric()
79 const uint32_t n_dims = getNumberOfDimensions(inputShape); in prepare()
83 NN_RET_CHECK_EQ(getNumberOfDimensions(beginShape), 1u); in prepare()
87 NN_RET_CHECK_EQ(getNumberOfDimensions(sizeShape), 1u); in prepare()
DL2Normalization.cpp53 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normFloat32Impl()
80 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8Impl()
112 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8SignedImpl()
139 int32_t ndim = getNumberOfDimensions(inputShape); in l2normFloat32()
168 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8()
184 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8Signed()
203 int32_t numDimensions = getNumberOfDimensions(input); in prepare()
DUnidirectionalSequenceRNN.cpp134 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3u); in prepare()
135 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2u); in prepare()
136 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentWeights), 2u); in prepare()
137 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1u); in prepare()
138 NN_RET_CHECK_EQ(getNumberOfDimensions(hiddenState), 2u); in prepare()
DSqueeze.cpp42 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in prepare()
44 NN_RET_CHECK_LE(getNumberOfDimensions(inputShape), 4u); in prepare()
48 NN_OPS_CHECK(getNumberOfDimensions(squeezeDimsShape) == 1); in prepare()
DSimpleMath.cpp58 int32_t* scratchBuffer = new int32_t[getNumberOfDimensions(inputShape)]; in meanGeneric()
73 getNumberOfDimensions(inputShape), outputData, in meanGeneric()
75 getNumberOfDimensions(outputShape), axis, axisSize, keepDims, scratchBuffer, in meanGeneric()
DSoftmax.cpp56 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxSlowFloat32()
84 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxFloat32()
131 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxQuant8Impl()
202 [[maybe_unused]] int32_t ndim = getNumberOfDimensions(inputShape); in softmaxQuant8()
236 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4u); in prepare()
DConcatenation.cpp65 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
90 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
149 uint32_t numDimensions = getNumberOfDimensions(input0); in prepare()
158 NN_RET_CHECK_EQ(getNumberOfDimensions(input), numDimensions); in prepare()
DGather.cpp36 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
61 output.dimensions.reserve(getNumberOfDimensions(input) + getNumberOfDimensions(indices) - 1); in prepare()
DLocalResponseNormalization.cpp53 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in localResponseNormFloat32Impl()
83 int32_t ndim = getNumberOfDimensions(inputShape); in localResponseNorm()
134 int32_t numDimensions = getNumberOfDimensions(input); in prepare()
DReverse.cpp44 const auto inputTensorRank = getNumberOfDimensions(inputShape); in prepare()
50 NN_RET_CHECK_EQ(getNumberOfDimensions(axisShape), 1U); in prepare()
DReduce.cpp49 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute()
67 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
73 NN_RET_CHECK_EQ(getNumberOfDimensions(axesShape), 1u); in prepare()
DDequantize.cpp51 for (int i = getNumberOfDimensions(inputShape) - 1; i > channelDim; --i) { in computePerChannel()
76 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4u); in prepare()
/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp250 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()
[all …]
/packages/modules/NeuralNetworks/common/types/operations/src/
DMirrorPad.cpp42 NN_RET_CHECK_EQ(getNumberOfDimensions(inputPaddingTensorShape), 2U) in validate()
70 auto inputTensorRank = getNumberOfDimensions(inputTensorShape); in validate()
78 auto inputTensorRank = getNumberOfDimensions(inputTensorShape); in validate()
79 auto outputTensorRank = getNumberOfDimensions(outputTensorShape); in validate()
DFullyConnected.cpp38 NN_RET_CHECK_GE(getNumberOfDimensions(input), 2u); in validateShapes()
39 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4u); in validateShapes()
40 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2u); in validateShapes()
41 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1u); in validateShapes()
DReverse.cpp42 NN_RET_CHECK_EQ(getNumberOfDimensions(inputAxisTensorShape), 1U) in validate()
64 auto inputTensorRank = getNumberOfDimensions(inputTensorShape); in validate()
65 auto outputTensorRank = getNumberOfDimensions(outputTensorShape); in validate()
DReduce.cpp36 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4u); in validateProdSum()
59 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4u); in validateMaxMin()
75 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4u); in validateLogical()
DBroadcast.cpp63 NN_RET_CHECK_LE(getNumberOfDimensions(input1), 4u); in validate()
64 NN_RET_CHECK_LE(getNumberOfDimensions(input2), 4u); in validate()
DConv2D.cpp31 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
32 const auto filterRank = getNumberOfDimensions(context->getInputShape(kFilterTensor)); in validate()
/packages/modules/NeuralNetworks/common/include/
DCpuOperationUtils.h120 NN_RET_CHECK_EQ(getNumberOfDimensions(nchwShape), 4u) in convertNchwToNhwc()
141 NN_RET_CHECK_EQ(getNumberOfDimensions(nhwcShape), 4u) in convertNhwcToNchw()
187 NN_RET_CHECK_EQ(getNumberOfDimensions(shape), 4u); in initialize()

123