Home
last modified time | relevance | path

Searched refs:NN_CHECK (Results 1 – 17 of 17) sorted by relevance

/packages/modules/NeuralNetworks/common/operations/
DMaximumMinimum.cpp45 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
47 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalGeneric()
49 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalGeneric()
54 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
71 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalQuant8()
73 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalQuant8()
75 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalQuant8()
82 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalQuant8()
91 NN_CHECK(in1.type == in2.type); in prepare()
DPow.cpp44 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
46 NN_CHECK(baseShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &baseFlatIndex)); in evalGeneric()
48 NN_CHECK(exponentShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &exponentFlatIndex)); in evalGeneric()
53 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
DLSTM.cpp155 NN_CHECK(params->cell_clip >= 0); in CheckInputTensorDimensions()
156 NN_CHECK(params->proj_clip >= 0); in CheckInputTensorDimensions()
191 NN_CHECK(cifg_weights_all_or_none); in CheckInputTensorDimensions()
215 NN_CHECK(peephole_weights_all_or_none); in CheckInputTensorDimensions()
229 NN_CHECK(IsNullInput(input_gate_bias)); in CheckInputTensorDimensions()
262 NN_CHECK(projecton_tensors_consistent == true); in CheckInputTensorDimensions()
307 NN_CHECK(NumInputsWithValues(operation, operands) >= 15 && in Prepare()
347 NN_CHECK(NumDimensions(input_) > 1); in Prepare()
430 NN_CHECK(inputRank == 2 || inputRank == 3); in LSTMEvalFloat32()
550 NN_CHECK(inputRank == 2 || inputRank == 3); in LSTMEvalFloat16()
DLSHProjection.cpp57 NN_CHECK(SizeOfDimension(hash, 1) <= 32); in Prepare()
60 NN_CHECK(NumDimensions(input) >= 1); in Prepare()
68 NN_CHECK(NumInputsWithValues(operation, operands) == 3); in Prepare()
DL2Normalization.cpp142 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normFloat32()
171 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8()
187 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8Signed()
DExpandDims.cpp28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
DSoftmax.cpp88 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxFloat32()
206 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxQuant8()
DArgMinMax.cpp54 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in argMinMaxGeneric()
DSplit.cpp32 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in splitGeneric()
DQuantizedLSTM.cpp327 NN_CHECK(tflite::CheckedLog2(prevCellState->scale, &stateScaleLog2Rounded)); in prepare()
330 NN_CHECK(stateIntegerBits == 4); in prepare()
DSVDF.cpp55 NN_CHECK(num_inputs == 6 || num_inputs == 7); in Prepare()
DLocalResponseNormalization.cpp90 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in localResponseNorm()
DRNN.cpp50 NN_CHECK(num_inputs == 6); in Prepare()
DBidirectionalSequenceLSTM.cpp267 NN_CHECK(NumDimensions(input_) == 3); in Prepare()
/packages/modules/NeuralNetworks/common/
DIndexedShapeWrapper.cpp37 NN_CHECK(isValid(*index)); in nextIndexInplace()
63 NN_CHECK(isValid(index)); in indexToFlatIndex()
74 NN_CHECK(index.size() >= strides.size()); in broadcastedIndexToFlatIndex()
80 NN_CHECK(currentIndex < currentDimSize || currentDimSize == 1); in broadcastedIndexToFlatIndex()
DOperationsUtils.cpp168 NN_CHECK(-numberOfDimensions <= *axis && *axis < numberOfDimensions); in handleNegativeAxis()
670 NN_CHECK(handleNegativeAxis(input, &axis)); in argMinMaxPrepare()
691 NN_CHECK(handleNegativeAxis(input, &axis)); in splitPrepare()
/packages/modules/NeuralNetworks/common/include/
DOperationsUtils.h31 #define NN_CHECK(x) NN_RET_CHECK(x) macro