Lines Matching refs:NN_OPS_CHECK

220     NN_OPS_CHECK(double_multiplier >= 0.);  in QuantizeMultiplierSmallerThanOne()
221 NN_OPS_CHECK(double_multiplier < 1.); in QuantizeMultiplierSmallerThanOne()
227 NN_OPS_CHECK(double_multiplier > 0.); in QuantizeMultiplierSmallerThanOne()
231 NN_OPS_CHECK(q_fixed <= (1LL << 31)); in QuantizeMultiplierSmallerThanOne()
236 NN_OPS_CHECK(*right_shift >= 0); in QuantizeMultiplierSmallerThanOne()
237 NN_OPS_CHECK(q_fixed <= std::numeric_limits<int32_t>::max()); in QuantizeMultiplierSmallerThanOne()
244 NN_OPS_CHECK(double_multiplier > 1.); in QuantizeMultiplierGreaterThanOne()
247 NN_OPS_CHECK(q_fixed <= (1LL << 31)); in QuantizeMultiplierGreaterThanOne()
252 NN_OPS_CHECK(*left_shift >= 0); in QuantizeMultiplierGreaterThanOne()
253 NN_OPS_CHECK(q_fixed <= std::numeric_limits<int32_t>::max()); in QuantizeMultiplierGreaterThanOne()
266 NN_OPS_CHECK(std::abs(input_product_scale - bias_scale) <= in GetQuantizedConvolutionMultipler()
268 NN_OPS_CHECK(input_product_scale >= 0); in GetQuantizedConvolutionMultipler()
399 NN_OPS_CHECK(strechDim == -1); in reshapePrepare()
412 NN_OPS_CHECK(numInputElements == numOutputElements); in reshapePrepare()
423 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthToSpacePrepare()
424 NN_OPS_CHECK(blockSize > 0); in depthToSpacePrepare()
431 NN_OPS_CHECK(channels % (blockSize * blockSize) == 0); in depthToSpacePrepare()
442 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToDepthPrepare()
443 NN_OPS_CHECK(blockSize > 0); in spaceToDepthPrepare()
450 NN_OPS_CHECK(height % blockSize == 0); in spaceToDepthPrepare()
451 NN_OPS_CHECK(width % blockSize == 0); in spaceToDepthPrepare()
463 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
464 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
484 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
485 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
486 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 1); in hashtableLookupPrepare()
512 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in padPrepare()
513 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
514 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
515 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in padPrepare()
522 NN_OPS_CHECK(beforePadding >= 0 && afterPadding >= 0); in padPrepare()
536 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in batchToSpacePrepare()
539 NN_OPS_CHECK(blockSizeShape.type == OperandType::TENSOR_INT32); in batchToSpacePrepare()
540 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in batchToSpacePrepare()
542 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in batchToSpacePrepare()
549 NN_OPS_CHECK(batches % (blockSizeData[0] * blockSizeData[1]) == 0); in batchToSpacePrepare()
563 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToBatchPrepare()
566 NN_OPS_CHECK(blockSizeShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
567 NN_OPS_CHECK(getNumberOfDimensions(blockSizeShape) == 1); in spaceToBatchPrepare()
569 NN_OPS_CHECK(getSizeOfDimension(blockSizeShape, 0) == 2); in spaceToBatchPrepare()
572 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
573 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
574 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == 2); in spaceToBatchPrepare()
575 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in spaceToBatchPrepare()
585 NN_OPS_CHECK(paddedHeight % blockSizeData[0] == 0); in spaceToBatchPrepare()
586 NN_OPS_CHECK(paddedWidth % blockSizeData[1] == 0); in spaceToBatchPrepare()
601 NN_OPS_CHECK(axisShape.type == OperandType::TENSOR_INT32); in meanPrepare()
602 NN_OPS_CHECK(getNumberOfDimensions(axisShape) == 1); in meanPrepare()
633 NN_OPS_CHECK(current >= 0 && current < numInputDims); in meanPrepare()
700 NN_OPS_CHECK(sizeOfAxisToSplit % numOutputs == 0); in splitPrepare()
718 NN_OPS_CHECK(input.type == OperandType::TENSOR_QUANT8_ASYMM || in groupedConvPrepare()
721 NN_OPS_CHECK(input.type == filter.type); in groupedConvPrepare()
725 NN_OPS_CHECK(bias.type == OperandType::TENSOR_INT32); in groupedConvPrepare()
727 NN_OPS_CHECK(input.type == bias.type); in groupedConvPrepare()
729 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in groupedConvPrepare()
730 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in groupedConvPrepare()
731 NN_OPS_CHECK(getNumberOfDimensions(bias) == 1); in groupedConvPrepare()
733 NN_OPS_CHECK(getSizeOfDimension(filter, 0) == getSizeOfDimension(bias, 0)); in groupedConvPrepare()
735 NN_OPS_CHECK(getSizeOfDimension(filter, 3) * numGroups == getSizeOfDimension(input, 3)); in groupedConvPrepare()
736 NN_OPS_CHECK(getSizeOfDimension(filter, 0) % numGroups == 0); in groupedConvPrepare()