Home
last modified time | relevance | path

Searched refs:outputTensorRank (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/types/operations/src/
DReverse.cpp65 auto outputTensorRank = getNumberOfDimensions(outputTensorShape); in validate() local
66 NN_RET_CHECK(!inputTensorRank || !outputTensorRank || inputTensorRank == outputTensorRank) in validate()
68 << "output tensor rank " << outputTensorRank; in validate()
DMirrorPad.cpp79 auto outputTensorRank = getNumberOfDimensions(outputTensorShape); in validate() local
80 NN_RET_CHECK_EQ(inputTensorRank, outputTensorRank) in validate()
83 << "output tensor rank " << outputTensorRank; in validate()
/packages/modules/NeuralNetworks/common/cpu_operations/
DPack.cpp62 const uint32_t outputTensorRank = firstInputTensorRank + 1; in prepare() local
71 outputShape.dimensions.resize(outputTensorRank); in prepare()
76 for (int32_t j = axisDimension + 1; j < int32_t(outputTensorRank); ++j) { in prepare()