Searched refs:NN_CHECK (Results 1 – 18 of 18) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | MaximumMinimum.cpp | 41 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
43 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalGeneric()
45 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalGeneric()
50 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
67 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalQuant8()
69 NN_CHECK(aShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &aFlatIndex)); in evalQuant8()
71 NN_CHECK(bShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &bFlatIndex)); in evalQuant8()
78 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalQuant8()
87 NN_CHECK(in1.type == in2.type); in prepare()
|
| D | Pow.cpp | 42 NN_CHECK(outputShapeIndexed.indexToFlatIndex(curIndex, &outputFlatIndex)); in evalGeneric()
44 NN_CHECK(baseShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &baseFlatIndex)); in evalGeneric()
46 NN_CHECK(exponentShapeIndexed.broadcastedIndexToFlatIndex(curIndex, &exponentFlatIndex)); in evalGeneric()
51 NN_CHECK(outputShapeIndexed.nextIndexInplace(&curIndex, &lastIndex)); in evalGeneric()
|
| D | LSHProjection.cpp | 43 NN_CHECK(num_inputs == 3 || num_inputs == 4); in Prepare()
49 NN_CHECK(SizeOfDimension(hash, 1) <= 32); in Prepare()
52 NN_CHECK(NumDimensions(input) >= 1); in Prepare()
59 NN_CHECK(NumInputsWithValues(operation, operands) == 3); in Prepare()
|
| D | TopK_V2.cpp | 57 NN_CHECK(k > 0); in prepare()
58 NN_CHECK(k <= input.dimensions.back()); in prepare()
|
| D | ExpandDims.cpp | 28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
|
| D | LSTM.cpp | 146 NN_CHECK(params->cell_clip >= 0); in CheckInputTensorDimensions()
147 NN_CHECK(params->proj_clip >= 0); in CheckInputTensorDimensions()
182 NN_CHECK(cifg_weights_all_or_none); in CheckInputTensorDimensions()
206 NN_CHECK(peephole_weights_all_or_none); in CheckInputTensorDimensions()
220 NN_CHECK(IsNullInput(input_gate_bias)); in CheckInputTensorDimensions()
253 NN_CHECK(projecton_tensors_consistent == true); in CheckInputTensorDimensions()
298 NN_CHECK(NumInputsWithValues(operation, operands) >= 15 && in Prepare()
304 NN_CHECK(NumDimensions(input_) > 1); in Prepare()
387 NN_CHECK(inputRank == 2 || inputRank == 3); in LSTMEvalFloat32()
507 NN_CHECK(inputRank == 2 || inputRank == 3); in LSTMEvalFloat16()
|
| D | ArgMinMax.cpp | 59 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in argMinMaxGeneric()
|
| D | Split.cpp | 31 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in splitGeneric()
|
| D | L2Normalization.cpp | 101 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normFloat32()
130 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8()
|
| D | Normalization.cpp | 78 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in localResponseNormFloat32()
|
| D | Softmax.cpp | 77 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxFloat32()
189 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxQuant8()
|
| D | QuantizedLSTM.cpp | 325 NN_CHECK(tflite::CheckedLog2(prevCellState->scale, &stateScaleLog2Rounded)); in prepare()
328 NN_CHECK(stateIntegerBits == 4); in prepare()
|
| D | RNN.cpp | 51 NN_CHECK(num_inputs == 5 || num_inputs == 6); in Prepare()
|
| D | SVDF.cpp | 53 NN_CHECK(num_inputs == 6 || num_inputs == 7); in Prepare()
|
| D | BidirectionalSequenceLSTM.cpp | 165 NN_CHECK(NumDimensions(input_) == 3); in Prepare()
205 NN_CHECK(aux_inputs_all_or_none); in Prepare()
|
| /frameworks/ml/nn/common/ |
| D | IndexedShapeWrapper.cpp | 33 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()
|
| D | OperationsUtils.cpp | 146 NN_CHECK(-numberOfDimensions <= *axis && *axis < numberOfDimensions); in handleNegativeAxis()
831 NN_CHECK(handleNegativeAxis(input, &axis)); in argMinMaxPrepare()
850 NN_CHECK(handleNegativeAxis(input, &axis)); in splitPrepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 29 #define NN_CHECK(x) NN_RET_CHECK(x) macro
|