Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DMaximumMinimum.cpp47 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
49 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalGeneric()
51 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalGeneric()
56 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
73 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalQuant8()
75 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalQuant8()
77 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalQuant8()
84 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalQuant8()
93 NN_CHECK(in1.type == in2.type); in prepare()
DPow.cpp45 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
47 NN_CHECK(baseShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &baseFlatIndex)); in evalGeneric()
49 NN_CHECK(exponentShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &exponentFlatIndex)); in evalGeneric()
54 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
DLSHProjection.cpp58 NN_CHECK(SizeOfDimension(hash, 1) <= 32); in Prepare()
61 NN_CHECK(NumDimensions(input) >= 1); in Prepare()
69 NN_CHECK(NumInputsWithValues(operation, operands) == 3); in Prepare()
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()
DExpandDims.cpp28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
DL2Normalization.cpp141 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normFloat32()
170 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8()
186 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8Signed()
DArgMinMax.cpp58 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in argMinMaxGeneric()
DSplit.cpp33 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in splitGeneric()
DSoftmax.cpp86 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxFloat32()
204 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxQuant8()
DQuantizedLSTM.cpp330 NN_CHECK(tflite::CheckedLog2(prevCellState->scale, &stateScaleLog2Rounded)); in prepare()
333 NN_CHECK(stateIntegerBits == 4); in prepare()
DLocalResponseNormalization.cpp89 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in localResponseNorm()
DSVDF.cpp57 NN_CHECK(num_inputs == 6 || num_inputs == 7); in Prepare()
DRNN.cpp54 NN_CHECK(num_inputs == 6); in Prepare()
DBidirectionalSequenceLSTM.cpp268 NN_CHECK(NumDimensions(input_) == 3); in Prepare()
/frameworks/ml/nn/common/
DIndexedShapeWrapper.cpp33 NN_CHECK(isValid(*index)); in nextIndexInplace()
59 NN_CHECK(isValid(index)); in indexToFlatIndex()
70 NN_CHECK(index.size() >= strides.size()); in broadcastedIndexToFlatIndex()
76 NN_CHECK(currentIndex < currentDimSize || currentDimSize == 1); in broadcastedIndexToFlatIndex()
DOperationsUtils.cpp170 NN_CHECK(-numberOfDimensions <= *axis && *axis < numberOfDimensions); in handleNegativeAxis()
672 NN_CHECK(handleNegativeAxis(input, &axis)); in argMinMaxPrepare()
693 NN_CHECK(handleNegativeAxis(input, &axis)); in splitPrepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h31 #define NN_CHECK(x) NN_RET_CHECK(x) macro