Searched refs:paddingShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Reshape.cpp | 230 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric() argument
237 blockSizeDim, blockSize, convertShapeToTflshape(paddingShape), padding, in spaceToBatchGeneric()
243 const Shape& paddingShape, float* outputData,
247 const Shape& paddingShape, _Float16* outputData,
251 const Shape& paddingShape, uint8_t* outputData,
255 const Shape& paddingShape, int8_t* outputData,
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 93 const int32_t* padding, const Shape& paddingShape, T* outputData,
|