Home
last modified time | relevance | path

Searched refs:NN_OPS_CHECK (Results 1 – 7 of 7) sorted by relevance

/frameworks/ml/nn/common/
DOperationsUtils.cpp220 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()
[all …]
/frameworks/ml/nn/common/operations/
DStridedSlice.cpp142 NN_OPS_CHECK(numInputDims <= 4); in prepare()
148 NN_OPS_CHECK(getNumberOfDimensions(beginShape) == 1); in prepare()
149 NN_OPS_CHECK(getNumberOfDimensions(endShape) == 1); in prepare()
150 NN_OPS_CHECK(getNumberOfDimensions(stridesShape) == 1); in prepare()
152 NN_OPS_CHECK(getSizeOfDimension(beginShape, 0) == numInputDims); in prepare()
153 NN_OPS_CHECK(getSizeOfDimension(endShape, 0) == numInputDims); in prepare()
154 NN_OPS_CHECK(getSizeOfDimension(stridesShape, 0) == numInputDims); in prepare()
156 NN_OPS_CHECK(beginShape.type == OperandType::TENSOR_INT32); in prepare()
157 NN_OPS_CHECK(endShape.type == OperandType::TENSOR_INT32); in prepare()
158 NN_OPS_CHECK(stridesShape.type == OperandType::TENSOR_INT32); in prepare()
[all …]
DSqueeze.cpp85 NN_OPS_CHECK(squeezeDimsShape.type == OperandType::TENSOR_INT32); in prepare()
86 NN_OPS_CHECK(getNumberOfDimensions(squeezeDimsShape) == 1); in prepare()
104 NN_OPS_CHECK(current >= 0 && current < numInputDims && in prepare()
DPow.cpp63 NN_OPS_CHECK(baseShape.type == exponentShape.type); in prepare()
DReshape.cpp95 NN_OPS_CHECK(numInputDims <= 4); in padGeneric()
DDepthwiseConv2D.cpp545 NN_OPS_CHECK(param.depth_multiplier * channels_in == channels_out); in prepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h32 #define NN_OPS_CHECK(x) NN_RET_CHECK(x) macro