Searched refs:outputKeypoint (Results 1 – 1 of 1) sorted by relevance
249 Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor); in prepare() local250 outputKeypoint.type = boxesShape.type; in prepare()251 outputKeypoint.dimensions = {numBoxes, numKeypoints, 2}; in prepare()252 outputKeypoint.offset = 0; in prepare()253 outputKeypoint.scale = 0.f; in prepare()256 outputKeypoint.scale = 0.125f; in prepare()258 NN_RET_CHECK(context->setOutputShape(kOutputKeypointTensor, outputKeypoint)); in prepare()