Searched refs:paddingShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Reshape.cpp | 210 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric() argument
217 blockSizeDim, blockSize, convertShapeToTflshape(paddingShape), padding, in spaceToBatchGeneric()
223 const Shape& paddingShape, float* outputData,
227 const Shape& paddingShape, _Float16* outputData,
231 const Shape& paddingShape, uint8_t* outputData,
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 107 const int32_t* padding, const Shape& paddingShape, T* outputData,
|