/frameworks/ml/nn/common/operations/ |
D | Broadcast.cpp | 183 op_params, convertShapeToTflshape(shape1), in1, convertShapeToTflshape(shape2), in addQuant8() 184 in2, convertShapeToTflshape(shapeOut), out); in addQuant8() 187 tflite::reference_ops::BroadcastAdd4DSlow(op_params, convertShapeToTflshape(shape1), in addQuant8() 188 in1, convertShapeToTflshape(shape2), in2, in addQuant8() 189 convertShapeToTflshape(shapeOut), out); in addQuant8() 194 tflite::optimized_integer_ops::Add(op_params, convertShapeToTflshape(shape1), in1, in addQuant8() 195 convertShapeToTflshape(shape2), in2, in addQuant8() 196 convertShapeToTflshape(shapeOut), out); in addQuant8() 199 tflite::optimized_ops::Add(op_params, convertShapeToTflshape(shape1), in1, in addQuant8() 200 convertShapeToTflshape(shape2), in2, in addQuant8() [all …]
|
D | Pooling.cpp | 144 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc() 145 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc() 167 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc() 168 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc() 179 tflite::reference_integer_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), in averagePoolNhwc() 180 inputData, convertShapeToTflshape(outputShape), in averagePoolNhwc() 190 tflite::optimized_ops::L2Pool(op_params, convertShapeToTflshape(inputShape), inputData, in l2PoolNhwc() 191 convertShapeToTflshape(outputShape), outputData); in l2PoolNhwc() 212 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc() 213 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc() [all …]
|
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()
|
D | Activation.cpp | 181 tflite::optimized_ops::Tanh(inputData, convertShapeToTflshape(inputShape), inputShape.offset, in tanhQuant8() 183 convertShapeToTflshape(outputShape)); in tanhQuant8() 212 inputData, convertShapeToTflshape(inputShape), inputShape.offset, input_range_radius, in logisticQuant8() 213 input_multiplier, input_left_shift, outputData, convertShapeToTflshape(outputShape)); in logisticQuant8() 352 tflite::reference_ops::HardSwish(params, convertShapeToTflshape(inputShape), inputData, in hardSwishQuant() 353 convertShapeToTflshape(outputShape), outputData); in hardSwishQuant() 588 tflite::reference_ops::HardSwish(convertShapeToTflshape(inputShape), inputFloat.data(), in executeHardSwish() 589 convertShapeToTflshape(outputShape), in executeHardSwish() 596 convertShapeToTflshape(context->getInputShape(kInputTensor)), in executeHardSwish() 598 convertShapeToTflshape(context->getOutputShape(kOutputTensor)), in executeHardSwish()
|
D | FullyConnected.cpp | 176 params, convertShapeToTflshape(inputShape), inputData, in fullyConnectedQuant8() 177 convertShapeToTflshape(weightsShape), weightsData, convertShapeToTflshape(biasShape), in fullyConnectedQuant8() 178 biasData, convertShapeToTflshape(outputShape), outputData); in fullyConnectedQuant8()
|
D | L2Normalization.cpp | 146 tflite::optimized_ops::L2Normalization(param, convertShapeToTflshape(inputShape), inputData, in l2normFloat32() 147 convertShapeToTflshape(outputShape), outputData); in l2normFloat32() 175 tflite::optimized_ops::L2Normalization(param, convertShapeToTflshape(inputShape), inputData, in l2normQuant8() 176 convertShapeToTflshape(outputShape), outputData); in l2normQuant8()
|
D | DepthwiseConv2D.cpp | 157 tflite::reference_ops::DepthwiseConv(params, convertShapeToTflshape(inputShape), inputData, in depthwiseConvNhwc() 158 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc() 159 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc() 160 convertShapeToTflshape(outputShape), outputData); in depthwiseConvNhwc() 233 tflite::reference_ops::DepthwiseConv(params, convertShapeToTflshape(inputShape), inputData, in depthwiseConvNhwc() 234 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc() 235 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc() 236 convertShapeToTflshape(outputShape), outputData); in depthwiseConvNhwc()
|
D | ResizeImageOps.cpp | 124 convertShapeToTflshape(inputShape), inputData, convertShapeToTflshape(outDimShape), in resizeImageOpNhwc() 125 outDimData, convertShapeToTflshape(outputShape), outputData); in resizeImageOpNhwc()
|
D | LocalResponseNormalization.cpp | 96 param, convertShapeToTflshape(inputShape), inputData, in localResponseNorm() 97 convertShapeToTflshape(outputShape), outputData); in localResponseNorm()
|
D | Softmax.cpp | 91 tflite::optimized_ops::Softmax(param, convertShapeToTflshape(inputShape), inputData, in softmaxFloat32() 92 convertShapeToTflshape(outputShape), outputData); in softmaxFloat32()
|
D | Conv2D.cpp | 501 convertShapeToTflshape(inputShape), inputData, convertShapeToTflshape(filterShape), in convQuant8PerChannelNhwc() 502 filterData, convertShapeToTflshape(biasShape), biasData, in convQuant8PerChannelNhwc() 503 convertShapeToTflshape(outputShape), outputData); in convQuant8PerChannelNhwc()
|
/frameworks/ml/nn/common/include/ |
D | CpuOperationUtils.h | 55 inline tflite::RuntimeShape convertShapeToTflshape(const Shape& shape) { in convertShapeToTflshape() function
|