Home
last modified time | relevance | path

Searched refs:NN_RET_CHECK_EQ (Results 1 – 25 of 43) sorted by relevance

12

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp112 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
113 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
186 NN_RET_CHECK_EQ(inputRank, 3) << "Invalid input tensor rank: " << inputRank; in prepare()
193 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
194 NN_RET_CHECK_EQ(getSizeOfDimension(inputToOutputShape, 1), inputSize); in prepare()
198 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
199 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numCells); in prepare()
204 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2); in prepare()
205 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numCells); in prepare()
206 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 1), inputSize); in prepare()
[all …]
DBidirectionalSequenceRNN.cpp250 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
275 NN_RET_CHECK_EQ(context->getNumOutputs(), 1); in prepare()
277 NN_RET_CHECK_EQ(context->getNumOutputs(), 2); in prepare()
323 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
324 NN_RET_CHECK_EQ(getNumberOfDimensions(fwWeights), 2); in prepare()
325 NN_RET_CHECK_EQ(getNumberOfDimensions(fwRecurrentWeights), 2); in prepare()
326 NN_RET_CHECK_EQ(getNumberOfDimensions(fwBias), 1); in prepare()
327 NN_RET_CHECK_EQ(getNumberOfDimensions(fwHiddenState), 2); in prepare()
328 NN_RET_CHECK_EQ(getNumberOfDimensions(bwWeights), 2); in prepare()
329 NN_RET_CHECK_EQ(getNumberOfDimensions(bwRecurrentWeights), 2); in prepare()
[all …]
DUnidirectionalSequenceRNN.cpp117 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
118 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
147 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
148 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in prepare()
149 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentWeights), 2); in prepare()
150 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
151 NN_RET_CHECK_EQ(getNumberOfDimensions(hiddenState), 2); in prepare()
153 NN_RET_CHECK_EQ(inputSize, getSizeOfDimension(weights, 1)); in prepare()
154 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(bias, 0)); in prepare()
155 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(recurrentWeights, 0)); in prepare()
[all …]
DQuantizedLSTM.cpp246 NN_RET_CHECK_EQ(NumDimensions(input), 2); in prepare()
247 NN_RET_CHECK_EQ(input->scale, 1. / 128.0); in prepare()
248 NN_RET_CHECK_EQ(input->zeroPoint, 128); in prepare()
253 NN_RET_CHECK_EQ(NumDimensions(prevOutput), 2); in prepare()
254 NN_RET_CHECK_EQ(SizeOfDimension(prevOutput, 0), numBatches); in prepare()
255 NN_RET_CHECK_EQ(prevOutput->scale, 1. / 128.0); in prepare()
256 NN_RET_CHECK_EQ(prevOutput->zeroPoint, 128); in prepare()
265 NN_RET_CHECK_EQ(NumDimensions(weights), 2); in prepare()
266 NN_RET_CHECK_EQ(SizeOfDimension(weights, 0), outputSize); in prepare()
267 NN_RET_CHECK_EQ(SizeOfDimension(weights, 1), columns); in prepare()
[all …]
DSlice.cpp79 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
80 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
99 NN_RET_CHECK_EQ(getNumberOfDimensions(beginShape), 1); in prepare()
100 NN_RET_CHECK_EQ(getSizeOfDimension(beginShape, 0), n_dims); in prepare()
103 NN_RET_CHECK_EQ(getNumberOfDimensions(sizeShape), 1); in prepare()
104 NN_RET_CHECK_EQ(getSizeOfDimension(sizeShape, 0), n_dims); in prepare()
DTranspose.cpp70 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
71 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
99 NN_RET_CHECK_EQ(numInputDims, 2); in prepare()
110 NN_RET_CHECK_EQ(getNumberOfDimensions(permShape), 1); in prepare()
111 NN_RET_CHECK_EQ(numInputDims, getSizeOfDimension(permShape, 0)); in prepare()
DHeatmapMaxKeypoint.cpp203 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
204 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
228 NN_RET_CHECK_EQ(getNumberOfDimensions(heatmapShape), 4); in prepare()
229 NN_RET_CHECK_EQ(getNumberOfDimensions(boxesShape), 2); in prepare()
235 NN_RET_CHECK_EQ(getSizeOfDimension(heatmapShape, layout ? 3 : 1), heatmapSize); in prepare()
237 NN_RET_CHECK_EQ(getSizeOfDimension(boxesShape, 0), numBoxes); in prepare()
238 NN_RET_CHECK_EQ(boxInfoLength, 4); in prepare()
241 NN_RET_CHECK_EQ(boxesShape.scale, 0.125f); in prepare()
242 NN_RET_CHECK_EQ(boxesShape.offset, 0); in prepare()
DReduce.cpp67 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateProdSum()
68 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateProdSum()
80 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateMaxMin()
81 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateMaxMin()
94 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateLogical()
95 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateLogical()
112 NN_RET_CHECK_EQ(getNumberOfDimensions(axesShape), 1u); in prepare()
DGenerateProposals.cpp175 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
176 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
200 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2); in prepare()
201 NN_RET_CHECK_EQ(getNumberOfDimensions(bboxDeltasShape), 2); in prepare()
202 NN_RET_CHECK_EQ(getNumberOfDimensions(batchesShape), 1); in prepare()
203 NN_RET_CHECK_EQ(getNumberOfDimensions(imageInfoShape), 2); in prepare()
212 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), kRoiDim); in prepare()
213 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numRois); in prepare()
214 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
215 NN_RET_CHECK_EQ(getSizeOfDimension(batchesShape, 0), numRois); in prepare()
[all …]
DRoiPooling.cpp170 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
171 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
207 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
208 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2); in prepare()
215 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), 4); in prepare()
216 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
234 NN_RET_CHECK_EQ(roiShape.scale, 0.125f); in prepare()
235 NN_RET_CHECK_EQ(roiShape.offset, 0); in prepare()
DConcatenation.cpp103 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
119 NN_RET_CHECK_EQ(input.scale, output.scale); in validate()
120 NN_RET_CHECK_EQ(input.offset, output.offset); in validate()
139 NN_RET_CHECK_EQ(getNumberOfDimensions(input), numDimensions); in prepare()
145 NN_RET_CHECK_EQ(getSizeOfDimension(input0, d), getSizeOfDimension(input, d)); in prepare()
DFullyConnected.cpp142 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
143 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
209 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in prepare()
217 NN_RET_CHECK_EQ(getSizeOfDimension(bias, 0), num_units); in prepare()
218 NN_RET_CHECK_EQ(input_size * batch_size, input_n_elements); in prepare()
DLogicalNot.cpp45 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
46 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DRoiAlign.cpp331 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
332 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
372 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
373 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2); in prepare()
385 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), 4); in prepare()
386 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
409 NN_RET_CHECK_EQ(roiShape.scale, 0.125f); in prepare()
410 NN_RET_CHECK_EQ(roiShape.offset, 0); in prepare()
DNeg.cpp52 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
53 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DInstanceNormalization.cpp93 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
94 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
114 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
DTransposeConv2D.cpp68 NN_RET_CHECK_EQ(getNumberOfDimensions(context->getInputShape(3)), 1); in initialize()
69 NN_RET_CHECK_EQ(getSizeOfDimension(context->getInputShape(3), 0), 4); in initialize()
428 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
440 NN_RET_CHECK_EQ(context->getInputExtraParams(kFilterTensor).channelQuant().channelDim, in validate()
480 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
481 NN_RET_CHECK_EQ(getNumberOfDimensions(filter), 4); in prepare()
482 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
495 NN_RET_CHECK_EQ(channels_in, getSizeOfDimension(filter, 3)); in prepare()
496 NN_RET_CHECK_EQ(channels_out, getSizeOfDimension(bias, 0)); in prepare()
DQuantize.cpp66 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
67 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DLogicalAndOr.cpp62 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
63 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DElementwise.cpp65 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
66 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DSelect.cpp70 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
71 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DChannelShuffle.cpp62 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
63 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp49 NN_RET_CHECK_EQ(model.extensionNameToPrefix.size(), 1u); // Assumes no other extensions in use. in getFibonacciExtensionPrefix()
50 NN_RET_CHECK_EQ(model.extensionNameToPrefix[0].name, TEST_VENDOR_FIBONACCI_EXTENSION_NAME); in getFibonacciExtensionPrefix()
59 NN_RET_CHECK_EQ(operationType, (prefix << kLowBitsType) | TEST_VENDOR_FIBONACCI); in isFibonacciOperation()
65 NN_RET_CHECK_EQ(operation.inputs.size(), kNumInputs); in validate()
66 NN_RET_CHECK_EQ(operation.outputs.size(), kNumOutputs); in validate()
/frameworks/ml/nn/common/include/
DCpuOperationUtils.h101 NN_RET_CHECK_EQ(getNumberOfDimensions(nchwShape), 4) in convertNchwToNhwc()
122 NN_RET_CHECK_EQ(getNumberOfDimensions(nhwcShape), 4) in convertNhwcToNchw()
168 NN_RET_CHECK_EQ(getNumberOfDimensions(shape), 4); in initialize()
DOperationsUtils.h33 #define NN_CHECK_EQ(x, y) NN_RET_CHECK_EQ(x, y)
410 NN_RET_CHECK_EQ(dimsA.size(), 3u); in mergeThirdDimension()
411 NN_RET_CHECK_EQ(dimsB.size(), 3u); in mergeThirdDimension()
413 NN_RET_CHECK_EQ(dimsA[0], dimsB[0]); in mergeThirdDimension()
414 NN_RET_CHECK_EQ(dimsA[1], dimsB[1]); in mergeThirdDimension()

12