Searched refs:paddingsShape (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
512 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in padPrepare()
513 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
514 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
515 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in padPrepare()
561 const Shape& paddingsShape, Shape* output) { in spaceToBatchPrepare() argument
572 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
573 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
574 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == 2); in spaceToBatchPrepare()
575 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in spaceToBatchPrepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 318 bool padPrepare(const Shape& input, const int32_t* paddingsData, const Shape& paddingsShape,
326 const Shape& paddingsShape, Shape* output);
|