Searched refs:permShape (Results 1 – 1 of 1) sorted by relevance
49 const Shape& permShape, T* outputData, const Shape& outputShape) { in transposeGeneric() argument56 int32_t permSize = perm == nullptr ? 2 : static_cast<int32_t>(getSizeOfDimension(permShape, 0)); in transposeGeneric()119 const Shape& permShape = context->getInputShape(kPermTensor); in prepare() local126 NN_RET_CHECK(permShape.type == OperandType::TENSOR_INT32); in prepare()127 NN_RET_CHECK_EQ(getNumberOfDimensions(permShape), 1); in prepare()128 NN_RET_CHECK_EQ(numInputDims, getSizeOfDimension(permShape, 0)); in prepare()