Home
last modified time | relevance | path

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

123

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp119 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
199 NN_RET_CHECK_EQ(inputRank, 3) << "Invalid input tensor rank: " << inputRank; in prepare()
206 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
207 NN_RET_CHECK_EQ(getSizeOfDimension(inputToOutputShape, 1), inputSize); in prepare()
211 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
212 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numCells); in prepare()
217 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2); in prepare()
218 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numCells); in prepare()
219 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 1), inputSize); in prepare()
223 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2); in prepare()
[all …]
DQLSTM.cpp107 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
108 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
180 NN_RET_CHECK_EQ(inputRank, 2) << "Invalid input tensor rank: " << inputRank; in prepare()
186 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
187 NN_RET_CHECK_EQ(getSizeOfDimension(inputToOutputShape, 1), inputSize); in prepare()
191 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
192 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numUnits); in prepare()
197 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2); in prepare()
198 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numUnits); in prepare()
199 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 1), inputSize); in prepare()
[all …]
DBidirectionalSequenceRNN.cpp320 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
394 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
395 NN_RET_CHECK_EQ(getNumberOfDimensions(fwWeights), 2); in prepare()
396 NN_RET_CHECK_EQ(getNumberOfDimensions(fwRecurrentWeights), 2); in prepare()
397 NN_RET_CHECK_EQ(getNumberOfDimensions(fwBias), 1); in prepare()
398 NN_RET_CHECK_EQ(getNumberOfDimensions(fwHiddenState), 2); in prepare()
399 NN_RET_CHECK_EQ(getNumberOfDimensions(bwWeights), 2); in prepare()
400 NN_RET_CHECK_EQ(getNumberOfDimensions(bwRecurrentWeights), 2); in prepare()
401 NN_RET_CHECK_EQ(getNumberOfDimensions(bwBias), 1); in prepare()
402 NN_RET_CHECK_EQ(getNumberOfDimensions(bwHiddenState), 2); in prepare()
[all …]
DUnidirectionalSequenceRNN.cpp132 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
169 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
170 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in prepare()
171 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentWeights), 2); in prepare()
172 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
173 NN_RET_CHECK_EQ(getNumberOfDimensions(hiddenState), 2); in prepare()
175 NN_RET_CHECK_EQ(inputSize, getSizeOfDimension(weights, 1)); in prepare()
176 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(bias, 0)); in prepare()
177 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(recurrentWeights, 0)); in prepare()
178 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(recurrentWeights, 1)); in prepare()
[all …]
DQuantizedLSTM.cpp251 NN_RET_CHECK_EQ(NumDimensions(input), 2); in prepare()
252 NN_RET_CHECK_EQ(input->scale, 1. / 128.0); in prepare()
253 NN_RET_CHECK_EQ(input->zeroPoint, 128); in prepare()
258 NN_RET_CHECK_EQ(NumDimensions(prevOutput), 2); in prepare()
259 NN_RET_CHECK_EQ(SizeOfDimension(prevOutput, 0), numBatches); in prepare()
260 NN_RET_CHECK_EQ(prevOutput->scale, 1. / 128.0); in prepare()
261 NN_RET_CHECK_EQ(prevOutput->zeroPoint, 128); in prepare()
270 NN_RET_CHECK_EQ(NumDimensions(weights), 2); in prepare()
271 NN_RET_CHECK_EQ(SizeOfDimension(weights, 0), outputSize); in prepare()
272 NN_RET_CHECK_EQ(SizeOfDimension(weights, 1), columns); in prepare()
[all …]
DBidirectionalSequenceLSTM.cpp346 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_input_weights_), 2); in Prepare()
347 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 0), n_fw_cell); in Prepare()
348 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()
350 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_forget_weights_), 2); in Prepare()
351 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_forget_weights_, 0), n_fw_cell); in Prepare()
352 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_forget_weights_, 1), n_aux_input); in Prepare()
354 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_cell_weights_), 2); in Prepare()
355 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 0), n_fw_cell); in Prepare()
356 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 1), n_aux_input); in Prepare()
358 NN_RET_CHECK_EQ(NumDimensions(fw_aux_input_to_output_weights_), 2); in Prepare()
[all …]
DElementwise.cpp89 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
90 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
101 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateAbs()
102 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateAbs()
114 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateFloor()
115 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateFloor()
DSlice.cpp85 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
86 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
111 NN_RET_CHECK_EQ(getNumberOfDimensions(beginShape), 1); in prepare()
112 NN_RET_CHECK_EQ(getSizeOfDimension(beginShape, 0), n_dims); in prepare()
115 NN_RET_CHECK_EQ(getNumberOfDimensions(sizeShape), 1); in prepare()
116 NN_RET_CHECK_EQ(getSizeOfDimension(sizeShape, 0), n_dims); in prepare()
DTranspose.cpp77 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
78 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
112 NN_RET_CHECK_EQ(numInputDims, 2); in prepare()
123 NN_RET_CHECK_EQ(getNumberOfDimensions(permShape), 1); in prepare()
124 NN_RET_CHECK_EQ(numInputDims, getSizeOfDimension(permShape, 0)); in prepare()
DReduce.cpp73 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateProdSum()
74 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateProdSum()
90 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateMaxMin()
91 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateMaxMin()
113 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validateLogical()
114 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateLogical()
136 NN_RET_CHECK_EQ(getNumberOfDimensions(axesShape), 1u); in prepare()
DGenerateProposals.cpp204 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
205 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
236 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2); in prepare()
237 NN_RET_CHECK_EQ(getNumberOfDimensions(bboxDeltasShape), 2); in prepare()
238 NN_RET_CHECK_EQ(getNumberOfDimensions(batchesShape), 1); in prepare()
239 NN_RET_CHECK_EQ(getNumberOfDimensions(imageInfoShape), 2); in prepare()
248 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), kRoiDim); in prepare()
249 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numRois); in prepare()
250 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
251 NN_RET_CHECK_EQ(getSizeOfDimension(batchesShape, 0), numRois); in prepare()
[all …]
DHeatmapMaxKeypoint.cpp231 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
232 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
263 NN_RET_CHECK_EQ(getNumberOfDimensions(heatmapShape), 4); in prepare()
264 NN_RET_CHECK_EQ(getNumberOfDimensions(boxesShape), 2); in prepare()
270 NN_RET_CHECK_EQ(getSizeOfDimension(heatmapShape, layout ? 3 : 1), heatmapSize); in prepare()
272 NN_RET_CHECK_EQ(getSizeOfDimension(boxesShape, 0), numBoxes); in prepare()
273 NN_RET_CHECK_EQ(boxInfoLength, 4); in prepare()
277 NN_RET_CHECK_EQ(boxesShape.scale, 0.125f); in prepare()
278 NN_RET_CHECK_EQ(boxesShape.offset, 0); in prepare()
DFill.cpp68 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
69 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
88 NN_RET_CHECK_EQ(getNumberOfDimensions(dimsShape), 1); in prepare()
DRoiPooling.cpp191 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
192 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
234 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
235 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2); in prepare()
242 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), 4); in prepare()
243 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
261 NN_RET_CHECK_EQ(roiShape.scale, 0.125f); in prepare()
262 NN_RET_CHECK_EQ(roiShape.offset, 0); in prepare()
DConcatenation.cpp144 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
162 NN_RET_CHECK_EQ(input.scale, output.scale); in validate()
163 NN_RET_CHECK_EQ(input.offset, output.offset); in validate()
189 NN_RET_CHECK_EQ(getNumberOfDimensions(input), numDimensions); in prepare()
195 NN_RET_CHECK_EQ(getSizeOfDimension(input0, d), getSizeOfDimension(input, d)); in prepare()
DFullyConnected.cpp198 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in validateShapes()
199 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in validateShapes()
206 NN_RET_CHECK_EQ(input_size * batch_size, input_n_elements); in validateShapes()
209 NN_RET_CHECK_EQ(bias_len, num_units); in validateShapes()
224 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
225 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DRank.cpp35 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
36 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DLogicalNot.cpp48 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
49 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DNeg.cpp54 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
55 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DElu.cpp59 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
60 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DInstanceNormalization.cpp106 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
107 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
127 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
DRoiAlign.cpp343 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
344 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
389 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
390 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2); in prepare()
402 NN_RET_CHECK_EQ(getSizeOfDimension(roiShape, 1), 4); in prepare()
403 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
426 NN_RET_CHECK_EQ(roiShape.scale, 0.125f); in prepare()
427 NN_RET_CHECK_EQ(roiShape.offset, 0); in prepare()
DTransposeConv2D.cpp78 NN_RET_CHECK_EQ(getNumberOfDimensions(context->getInputShape(3)), 1); in initialize()
79 NN_RET_CHECK_EQ(getSizeOfDimension(context->getInputShape(3), 0), 4); in initialize()
442 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
455 NN_RET_CHECK_EQ(context->getInputExtraParams(kFilterTensor).channelQuant().channelDim, in validate()
500 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
501 NN_RET_CHECK_EQ(getNumberOfDimensions(filter), 4); in prepare()
502 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
515 NN_RET_CHECK_EQ(channels_in, getSizeOfDimension(filter, 3)); in prepare()
516 NN_RET_CHECK_EQ(channels_out, getSizeOfDimension(bias, 0)); in prepare()
DLogicalAndOr.cpp64 NN_RET_CHECK_EQ(context->getNumInputs(), kNumInputs); in validate()
65 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp52 NN_RET_CHECK_EQ(model.extensionNameToPrefix.size(), 1u); // Assumes no other extensions in use. in getFibonacciExtensionPrefix()
53 NN_RET_CHECK_EQ(model.extensionNameToPrefix[0].name, EXAMPLE_FIBONACCI_EXTENSION_NAME); in getFibonacciExtensionPrefix()
62 NN_RET_CHECK_EQ(operationType, (prefix << kLowBitsType) | EXAMPLE_FIBONACCI); in isFibonacciOperation()
68 NN_RET_CHECK_EQ(operation.inputs.size(), kNumInputs); in validate()
69 NN_RET_CHECK_EQ(operation.outputs.size(), kNumOutputs); in validate()

123