Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DHeatmapMaxKeypoint.cpp288 Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor); in prepare() local
289 outputKeypoint.type = boxesShape.type; in prepare()
290 outputKeypoint.dimensions = {numBoxes, numKeypoints, 2}; in prepare()
291 outputKeypoint.offset = 0; in prepare()
292 outputKeypoint.scale = 0.f; in prepare()
295 outputKeypoint.scale = 0.125f; in prepare()
297 NN_RET_CHECK(context->setOutputShape(kOutputKeypointTensor, outputKeypoint)); in prepare()