Home
last modified time | relevance | path

Searched refs:paddingsShape (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/
DOperationsUtils.cpp571 const Shape& paddingsShape, in padPrepare() argument
578 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in padPrepare()
579 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in padPrepare()
580 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == numInputDims); in padPrepare()
581 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in padPrepare()
633 const Shape& paddingsShape, in spaceToBatchPrepare() argument
645 NN_OPS_CHECK(paddingsShape.type == OperandType::TENSOR_INT32); in spaceToBatchPrepare()
646 NN_OPS_CHECK(getNumberOfDimensions(paddingsShape) == 2); in spaceToBatchPrepare()
647 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 0) == 2); in spaceToBatchPrepare()
648 NN_OPS_CHECK(getSizeOfDimension(paddingsShape, 1) == 2); in spaceToBatchPrepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h245 const Shape& paddingsShape,
257 const Shape& paddingsShape,