Home
last modified time | relevance | path

Searched refs:convertShapeToTflshape (Results 1 – 12 of 12) sorted by relevance

/frameworks/ml/nn/common/operations/
DBroadcast.cpp183 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 …]
DPooling.cpp144 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 …]
DReshape.cpp112 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()
DActivation.cpp181 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()
DFullyConnected.cpp176 params, convertShapeToTflshape(inputShape), inputData, in fullyConnectedQuant8()
177 convertShapeToTflshape(weightsShape), weightsData, convertShapeToTflshape(biasShape), in fullyConnectedQuant8()
178 biasData, convertShapeToTflshape(outputShape), outputData); in fullyConnectedQuant8()
DL2Normalization.cpp146 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()
DDepthwiseConv2D.cpp157 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()
DResizeImageOps.cpp124 convertShapeToTflshape(inputShape), inputData, convertShapeToTflshape(outDimShape), in resizeImageOpNhwc()
125 outDimData, convertShapeToTflshape(outputShape), outputData); in resizeImageOpNhwc()
DLocalResponseNormalization.cpp96 param, convertShapeToTflshape(inputShape), inputData, in localResponseNorm()
97 convertShapeToTflshape(outputShape), outputData); in localResponseNorm()
DSoftmax.cpp91 tflite::optimized_ops::Softmax(param, convertShapeToTflshape(inputShape), inputData, in softmaxFloat32()
92 convertShapeToTflshape(outputShape), outputData); in softmaxFloat32()
DConv2D.cpp501 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/
DCpuOperationUtils.h55 inline tflite::RuntimeShape convertShapeToTflshape(const Shape& shape) { in convertShapeToTflshape() function