Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DTransposeConv2D.cpp80 const int32_t* outputShapeData = context->getInputBuffer<int32_t>(3); in initialize() local
81 int32_t outputWidth = useNchw ? outputShapeData[3] : outputShapeData[2]; in initialize()
82 int32_t outputHeight = useNchw ? outputShapeData[2] : outputShapeData[1]; in initialize()