| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | Broadcast.cpp | 186 op_params, convertShapeToTflshape(shape1), in1, convertShapeToTflshape(shape2), in addQuant8()
187 in2, convertShapeToTflshape(shapeOut), out); in addQuant8()
198 convertShapeToTflshape(shape2), in2, in addQuant8()
199 convertShapeToTflshape(shapeOut), out); in addQuant8()
202 tflite::optimized_ops::Add(op_params, convertShapeToTflshape(shape1), in1, in addQuant8()
203 convertShapeToTflshape(shape2), in2, in addQuant8()
204 convertShapeToTflshape(shapeOut), out); in addQuant8()
309 convertShapeToTflshape(shape2), in2, in mulQuant8()
310 convertShapeToTflshape(shapeOut), out); in mulQuant8()
405 convertShapeToTflshape(shape2), in2, in subQuant8()
[all …]
|
| H A D | Pooling.cpp | 147 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
170 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
181 tflite::reference_integer_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), in averagePoolNhwc()
182 inputData, convertShapeToTflshape(outputShape), in averagePoolNhwc()
192 tflite::optimized_ops::L2Pool(op_params, convertShapeToTflshape(inputShape), inputData, in l2PoolNhwc()
193 convertShapeToTflshape(outputShape), outputData); in l2PoolNhwc()
214 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
215 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
224 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
225 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
[all …]
|
| H A D | Activation.cpp | 182 tflite::optimized_ops::Tanh(inputData, convertShapeToTflshape(inputShape), inputShape.offset, in tanhQuant8()
184 convertShapeToTflshape(outputShape)); in tanhQuant8()
213 inputData, convertShapeToTflshape(inputShape), inputShape.offset, input_range_radius, in logisticQuant8()
214 input_multiplier, input_left_shift, outputData, convertShapeToTflshape(outputShape)); in logisticQuant8()
278 input_left_shift, convertShapeToTflshape(inputShape), in tanhQuant8Signed()
279 inputData, convertShapeToTflshape(outputShape), outputData); in tanhQuant8Signed()
354 tflite::reference_ops::HardSwish(params, convertShapeToTflshape(inputShape), inputData, in hardSwishQuant()
355 convertShapeToTflshape(outputShape), outputData); in hardSwishQuant()
599 convertShapeToTflshape(outputShape), in executeHardSwish()
606 convertShapeToTflshape(context->getInputShape(kInputTensor)), in executeHardSwish()
[all …]
|
| H A D | Reshape.cpp | 112 tflite::RuntimeShape::ExtendedShape(4, convertShapeToTflshape(inputShape)); in padGeneric()
114 tflite::RuntimeShape::ExtendedShape(4, convertShapeToTflshape(outputShape)); in padGeneric()
230 {.output_offset = outputShape.offset}, convertShapeToTflshape(inputShape), inputData, in spaceToBatchGeneric()
231 blockSizeDim, blockSize, convertShapeToTflshape(paddingShape), padding, in spaceToBatchGeneric()
232 convertShapeToTflshape(outputShape), outputData); in spaceToBatchGeneric()
|
| H A D | FullyConnected.cpp | 177 params, convertShapeToTflshape(inputShape), inputData, in fullyConnectedQuant8()
178 convertShapeToTflshape(weightsShape), weightsData, convertShapeToTflshape(biasShape), in fullyConnectedQuant8()
179 biasData, convertShapeToTflshape(outputShape), outputData); in fullyConnectedQuant8()
|
| H A D | L2Normalization.cpp | 147 tflite::optimized_ops::L2Normalization(param, convertShapeToTflshape(inputShape), inputData, in l2normFloat32()
148 convertShapeToTflshape(outputShape), outputData); in l2normFloat32()
176 tflite::optimized_ops::L2Normalization(param, convertShapeToTflshape(inputShape), inputData, in l2normQuant8()
177 convertShapeToTflshape(outputShape), outputData); in l2normQuant8()
|
| H A D | DepthwiseConv2D.cpp | 159 tflite::reference_ops::DepthwiseConv(params, convertShapeToTflshape(inputShape), inputData, in depthwiseConvNhwc()
160 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
161 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc()
162 convertShapeToTflshape(outputShape), outputData); in depthwiseConvNhwc()
235 tflite::reference_ops::DepthwiseConv(params, convertShapeToTflshape(inputShape), inputData, in depthwiseConvNhwc()
236 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
237 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc()
238 convertShapeToTflshape(outputShape), outputData); in depthwiseConvNhwc()
|
| H A D | ResizeImageOps.cpp | 126 convertShapeToTflshape(inputShape), inputData, convertShapeToTflshape(outDimShape), in resizeImageOpNhwc()
127 outDimData, convertShapeToTflshape(outputShape), outputData); in resizeImageOpNhwc()
|
| H A D | LocalResponseNormalization.cpp | 98 param, convertShapeToTflshape(inputShape), inputData,
99 convertShapeToTflshape(outputShape), outputData);
|
| H A D | Softmax.cpp | 93 tflite::optimized_ops::Softmax(param, convertShapeToTflshape(inputShape), inputData, in softmaxFloat32()
94 convertShapeToTflshape(outputShape), outputData); in softmaxFloat32()
|
| H A D | Conv2D.cpp | 502 convertShapeToTflshape(inputShape), inputData, convertShapeToTflshape(filterShape), in convQuant8PerChannelNhwc()
503 filterData, convertShapeToTflshape(biasShape), biasData, in convQuant8PerChannelNhwc()
504 convertShapeToTflshape(outputShape), outputData); in convQuant8PerChannelNhwc()
|
| /aosp12/packages/modules/NeuralNetworks/common/include/ |
| H A D | CpuOperationUtils.h | 56 inline tflite::RuntimeShape convertShapeToTflshape(const Shape& shape) { in convertShapeToTflshape() function
|