Searched refs:paddingShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| 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,
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 106 const int32_t* padding, const Shape& paddingShape, T* outputData,
|