Home
last modified time | relevance | path

Searched refs:getNumberOfElements (Results 1 – 25 of 50) sorted by relevance

12

/frameworks/ml/nn/common/operations/
DSimpleMath.cpp34 std::vector<float> inputDataFloat32(getNumberOfElements(shape)); in floorFloat16()
37 std::vector<float> outputDataFloat32(getNumberOfElements(shape)); in floorFloat16()
55 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in meanFloat16()
58 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in meanFloat16()
77 U* tempSumBuffer = new (std::nothrow) U[getNumberOfElements(outputShape)]; in meanGeneric()
DActivation.cpp43 int numElements = getNumberOfElements(inputShape); in reluFloat()
79 int numElements = getNumberOfElements(inputShape); in tanhFloat16()
89 int numElements = getNumberOfElements(inputShape); in tanhFloat32()
100 int numElements = getNumberOfElements(inputShape); in logisticFloat()
112 int numElements = getNumberOfElements(inputShape); \
155 int numElements = getNumberOfElements(inputShape); in tanhQuant8()
185 int numElements = getNumberOfElements(inputShape); in logisticQuant8()
256 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu()
280 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu1()
304 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeRelu6()
[all …]
DSVDF.cpp100 std::vector<float> inputDataFloat32(getNumberOfElements(input_->shape())); in Eval()
102 std::vector<float> inputStateDataFloat32(getNumberOfElements(state_in_->shape())); in Eval()
105 std::vector<float> biasDataFloat32(getNumberOfElements(bias_->shape())); in Eval()
111 getNumberOfElements(weights_feature_->shape())); in Eval()
114 std::vector<float> weightsTimeDataFloat32(getNumberOfElements(weights_time_->shape())); in Eval()
117 std::vector<float> outputDataFloat32(getNumberOfElements(output_->shape())); in Eval()
118 std::vector<float> outputStateDataFloat32(getNumberOfElements(state_out_->shape())); in Eval()
DNormalization.cpp34 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in localResponseNormFloat32Impl()
37 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in localResponseNormFloat32Impl()
63 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in localResponseNormFloat16()
65 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in localResponseNormFloat16()
DFullyConnected.cpp58 uint32_t input_n_elements = getNumberOfElements(inputShape); in fullyConnectedFloat32()
82 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in fullyConnectedFloat16()
84 std::vector<float> weightsDataFloat32(getNumberOfElements(weightsShape)); in fullyConnectedFloat16()
86 std::vector<float> biasDataFloat32(getNumberOfElements(biasShape)); in fullyConnectedFloat16()
89 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in fullyConnectedFloat16()
210 uint32_t input_n_elements = getNumberOfElements(input); in prepare()
228 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DHeatmapMaxKeypoint.cpp183 std::vector<float> heatmap_float32(getNumberOfElements(heatmapShape)); in heatmapMaxKeypointQuant()
185 std::vector<float> boxes_float32(getNumberOfElements(boxesShape)); in heatmapMaxKeypointQuant()
187 std::vector<float> outputScore_float32(getNumberOfElements(outputScoreShape)); in heatmapMaxKeypointQuant()
188 std::vector<float> outputKeypoint_float32(getNumberOfElements(outputKeypointShape)); in heatmapMaxKeypointQuant()
271 std::vector<float> heatmap_float32(getNumberOfElements(heatmapShape)); in execute()
273 std::vector<float> boxes_float32(getNumberOfElements(boxesShape)); in execute()
275 std::vector<float> outputScore_float32(getNumberOfElements(outputScoreShape)); in execute()
276 std::vector<float> outputKeypoint_float32(getNumberOfElements(outputKeypointShape)); in execute()
DQuantize.cpp42 uint32_t size = getNumberOfElements(outputShape); in quantizeFloat32ToQuant8()
54 uint32_t size = getNumberOfElements(outputShape); in quantizeFloat16ToQuant8()
89 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DL2Normalization.cpp42 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normFloat32Impl()
45 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normFloat32Impl()
69 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Impl()
72 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8Impl()
117 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in l2normFloat16()
119 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in l2normFloat16()
DSoftmax.cpp45 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in softmaxSlowFloat32()
48 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxSlowFloat32()
93 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in softmaxFloat16()
95 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in softmaxFloat16()
119 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in softmaxQuant8Impl()
122 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxQuant8Impl()
268 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DPooling.cpp134 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in averagePoolNhwc()
135 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in averagePoolNhwc()
167 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in l2PoolNhwc()
168 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in l2PoolNhwc()
199 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in maxPoolNhwc()
200 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in maxPoolNhwc()
340 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAveragePool()
354 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeL2Pool()
367 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMaxPool()
DArgMinMax.cpp33 const int outerSize = getNumberOfElements(inputShape, 0, axis); in argMinMaxImpl()
35 const int innerSize = getNumberOfElements( in argMinMaxImpl()
DDequantize.cpp38 const int numElements = getNumberOfElements(inputShape); in compute()
59 const int numElements = getNumberOfElements(inputShape); in computePerChannel()
108 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DGenerateProposals.cpp63 const float* roiDataEnd = roiData + getNumberOfElements(roiShape); in bboxTransformFloat32()
107 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in bboxTransformFloat16()
109 std::vector<float> delta_float32(getNumberOfElements(bboxDeltasShape)); in bboxTransformFloat16()
111 std::vector<float> imageInfo_float32(getNumberOfElements(imageInfoDataShape)); in bboxTransformFloat16()
113 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformFloat16()
127 std::vector<float> roi_float32(getNumberOfElements(roiShape)); in bboxTransformQuant()
129 std::vector<float> delta_float32(getNumberOfElements(bboxDeltasShape)); in bboxTransformQuant()
132 std::vector<float> imageInfo_float32(getNumberOfElements(imageInfoDataShape)); in bboxTransformQuant()
135 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformQuant()
236 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
[all …]
DGather.cpp43 const auto outerSize = getNumberOfElements(inputShape, 0, axis); in eval()
46 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
47 const auto indicesCount = getNumberOfElements(indicesShape); in eval()
DBroadcast.cpp71 std::vector<float> in1_float32(getNumberOfElements(shape1)); in binaryOperationFloat16()
73 std::vector<float> in2_float32(getNumberOfElements(shape2)); in binaryOperationFloat16()
75 std::vector<float> out_float32(getNumberOfElements(shapeOut)); in binaryOperationFloat16()
253 uint32_t numOutputElements = getNumberOfElements(shapeOut); in subFloat32()
394 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeAdd()
427 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeMul()
460 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeSub()
493 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in executeDiv()
DDepthwiseConv2D.cpp36 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in depthwiseConvFloat16()
38 std::vector<float> filterDataFloat32(getNumberOfElements(filterShape)); in depthwiseConvFloat16()
40 std::vector<float> biasDataFloat32(getNumberOfElements(biasShape)); in depthwiseConvFloat16()
43 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in depthwiseConvFloat16()
DTransposeConv2D.cpp125 memset(outputData, 0, getNumberOfElements(outputShape) * sizeof(float)); in transposeConvNhwc()
180 uint32_t tempBufferByteSize = getNumberOfElements(outputShape) * sizeof(int32_t); in transposeConvNhwc()
272 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in transposeConvNhwc()
273 std::vector<float> filterData_float32(getNumberOfElements(filterShape)); in transposeConvNhwc()
274 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in transposeConvNhwc()
275 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in transposeConvNhwc()
315 uint32_t tempBufferByteSize = getNumberOfElements(outputShape) * sizeof(int32_t); in transposeConvQuant8PerChannelNhwc()
523 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DChannelShuffle.cpp41 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in eval()
44 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
DLogSoftmax.cpp42 const uint32_t outerSize = getNumberOfElements(shape, 0, axis); in compute()
44 const uint32_t innerSize = getNumberOfElements(shape, axis + 1, getNumberOfDimensions(shape)); in compute()
DGroupedConv2D.cpp270 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in groupedConvFloat16()
271 std::vector<float> filterData_float32(getNumberOfElements(filterShape)); in groupedConvFloat16()
272 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in groupedConvFloat16()
273 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in groupedConvFloat16()
DResizeImageOps.cpp79 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in resizeImageOpNhwc()
81 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in resizeImageOpNhwc()
194 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
DLogicalNot.cpp35 const auto size = getNumberOfElements(shape); in compute()
DQuantizedLSTM.cpp378 std::vector<uint8_t> weights(getNumberOfElements(weightsShape)); in eval()
383 std::vector<int32_t> bias(getNumberOfElements(biasShape)); in eval()
393 std::vector<uint8_t> concatTemp(getNumberOfElements(concatTempShape)); in eval()
394 std::vector<int16_t> activationTemp(getNumberOfElements(activationTempShape)); in eval()
/frameworks/ml/nn/runtime/test/fuzzing/
DRandomGraphGeneratorUtils.cpp78 operand->getNumberOfElements()); in dump()
142 op->getNumberOfElements()); in dump()
DRandomGraphGenerator.cpp60 uint32_t RandomOperand::getNumberOfElements() const { in getNumberOfElements() function in android::nn::fuzzing_test::RandomOperand
67 return kSizeOfDataType[static_cast<int32_t>(dataType)] * getNumberOfElements(); in getBufferSize()
283 uint32_t len = op.getNumberOfElements(); in expectNear()
328 uint32_t len = op.getNumberOfElements(); in expectBooleanNearlyEqual()

12