Searched refs:paddingsShape (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
508 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in padPrepare()
509 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
510 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
511 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in padPrepare()
557 const Shape& paddingsShape, Shape* output) { in spaceToBatchPrepare() argument
568 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
569 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
570 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == 2); in spaceToBatchPrepare()
571 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);
|