Searched refs:convertShapeToTflshape (Results 1 – 9 of 9) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 90 tflite::optimized_ops::DepthwiseConv(params, convertShapeToTflshape(inputShape), inputData, in depthwiseConvFloat32()
91 convertShapeToTflshape(filterShape), filterData, in depthwiseConvFloat32()
92 convertShapeToTflshape(biasShape), biasData, in depthwiseConvFloat32()
93 convertShapeToTflshape(outputShape), outputData); in depthwiseConvFloat32()
139 tflite::optimized_ops::DepthwiseConv(params, convertShapeToTflshape(inputShape), inputData, in depthwiseConvQuant8()
140 convertShapeToTflshape(filterShape), filterData, in depthwiseConvQuant8()
141 convertShapeToTflshape(biasShape), biasData, in depthwiseConvQuant8()
142 convertShapeToTflshape(outputShape), outputData); in depthwiseConvQuant8()
|
| D | Pooling.cpp | 126 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc()
127 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
149 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc()
150 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
159 tflite::optimized_ops::L2Pool(op_params, convertShapeToTflshape(inputShape), inputData, in l2PoolNhwc()
160 convertShapeToTflshape(outputShape), outputData); in l2PoolNhwc()
181 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
182 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
191 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
192 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
|
| D | ResizeImageOps.cpp | 60 convertShapeToTflshape(inputShape), inputData, in resizeImageOpNhwc()
61 convertShapeToTflshape(outDimShape), outDimData, in resizeImageOpNhwc()
62 convertShapeToTflshape(outputShape), outputData); in resizeImageOpNhwc()
67 {.align_corners = false}, convertShapeToTflshape(inputShape), inputData, in resizeImageOpNhwc()
68 convertShapeToTflshape(outDimShape), outDimData, in resizeImageOpNhwc()
69 convertShapeToTflshape(outputShape), outputData); in resizeImageOpNhwc()
|
| D | Reshape.cpp | 104 tflite::RuntimeShape::ExtendedShape(4, convertShapeToTflshape(inputShape)); in padGeneric()
106 tflite::RuntimeShape::ExtendedShape(4, convertShapeToTflshape(outputShape)); in padGeneric()
216 {.output_offset = outputShape.offset}, convertShapeToTflshape(inputShape), inputData, in spaceToBatchGeneric()
217 blockSizeDim, blockSize, convertShapeToTflshape(paddingShape), padding, in spaceToBatchGeneric()
218 convertShapeToTflshape(outputShape), outputData); in spaceToBatchGeneric()
|
| D | L2Normalization.cpp | 106 tflite::optimized_ops::L2Normalization(param, convertShapeToTflshape(inputShape), inputData, in l2normFloat32()
107 convertShapeToTflshape(outputShape), outputData); in l2normFloat32()
135 tflite::optimized_ops::L2Normalization(param, convertShapeToTflshape(inputShape), inputData, in l2normQuant8()
136 convertShapeToTflshape(outputShape), outputData); in l2normQuant8()
|
| D | Normalization.cpp | 85 param, convertShapeToTflshape(inputShape), inputData, in localResponseNormFloat32()
86 convertShapeToTflshape(outputShape), outputData); in localResponseNormFloat32()
|
| D | Softmax.cpp | 82 tflite::optimized_ops::Softmax(param, convertShapeToTflshape(inputShape), inputData, in softmaxFloat32()
83 convertShapeToTflshape(outputShape), outputData); in softmaxFloat32()
215 tflite::optimized_ops::Softmax(param, convertShapeToTflshape(inputShape), inputData, in softmaxQuant8()
216 convertShapeToTflshape(outputShape), outputData); in softmaxQuant8()
|
| D | Activation.cpp | 170 tflite::optimized_ops::Tanh(inputData, convertShapeToTflshape(inputShape), inputShape.offset, in tanhQuant8()
172 convertShapeToTflshape(outputShape)); in tanhQuant8()
201 inputData, convertShapeToTflshape(inputShape), inputShape.offset, input_range_radius, in logisticQuant8()
202 input_multiplier, input_left_shift, outputData, convertShapeToTflshape(outputShape)); in logisticQuant8()
|
| /frameworks/ml/nn/common/include/ |
| D | CpuOperationUtils.h | 54 inline tflite::RuntimeShape convertShapeToTflshape(const Shape& shape) { in convertShapeToTflshape() function
|