Searched refs:paddingsData (Results 1 – 2 of 2) sorted by relevance
/frameworks/ml/nn/common/ |
D | OperationsUtils.cpp | 503 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape, in padPrepare() argument 515 int32_t beforePadding = *paddingsData++; in padPrepare() 516 int32_t afterPadding = *paddingsData++; in padPrepare() 556 const Shape& blockSizeShape, const int32_t* paddingsData, in spaceToBatchPrepare() argument 578 uint32_t paddedHeight = paddingsData[0] + height + paddingsData[1]; in spaceToBatchPrepare() 579 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,
|