Searched refs:paddingShape (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | Reshape.cpp | 224 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric() argument
231 blockSizeDim, blockSize, convertShapeToTflshape(paddingShape), padding, in spaceToBatchGeneric()
237 const Shape& paddingShape, float* outputData,
241 const Shape& paddingShape, _Float16* outputData,
245 const Shape& paddingShape, uint8_t* outputData,
249 const Shape& paddingShape, int8_t* outputData,
|
| /aosp12/packages/modules/NeuralNetworks/common/include/ |
| H A D | Operations.h | 110 const int32_t* padding, const Shape& paddingShape, T* outputData,
|