Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DTransposeConv2D.cpp70 const int32_t* outputShapeData = context->getInputBuffer<int32_t>(3); in initialize() local
71 int32_t outputWidth = useNchw ? outputShapeData[3] : outputShapeData[2]; in initialize()
72 int32_t outputHeight = useNchw ? outputShapeData[2] : outputShapeData[1]; in initialize()