Searched refs:paddingsShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 501 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape, in padPrepare() argument
506 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in padPrepare()
507 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
508 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
509 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in padPrepare()
555 const Shape& paddingsShape, Shape* output) { in spaceToBatchPrepare() argument
566 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
567 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
568 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == 2); in spaceToBatchPrepare()
569 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in spaceToBatchPrepare()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsUtils.h | 302 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape,
310 const Shape& paddingsShape, Shape* output);
|