Searched refs:paddingsShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 410 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape, in padPrepare() argument
415 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in padPrepare()
416 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
417 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
418 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in padPrepare()
464 const Shape& paddingsShape, Shape* output) { in spaceToBatchPrepare() argument
475 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
476 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
477 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == 2); in spaceToBatchPrepare()
478 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in spaceToBatchPrepare()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsExecutionUtils.h | 244 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape,
252 const Shape& paddingsShape, Shape* output);
|