Searched refs:paddingsData (Results 1 – 2 of 2) sorted by relevance
/frameworks/ml/nn/common/ |
D | OperationsUtils.cpp | 507 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape, in padPrepare() argument 519 int32_t beforePadding = *paddingsData++; in padPrepare() 520 int32_t afterPadding = *paddingsData++; in padPrepare() 560 const Shape& blockSizeShape, const int32_t* paddingsData, in spaceToBatchPrepare() argument 582 uint32_t paddedHeight = paddingsData[0] + height + paddingsData[1]; in spaceToBatchPrepare() 583 uint32_t paddedWidth = paddingsData[2] + width + paddingsData[3]; in spaceToBatchPrepare()
|
/frameworks/ml/nn/common/include/ |
D | OperationsUtils.h | 318 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape, 325 const Shape& blockSizeShape, const int32_t* paddingsData,
|