Searched defs:classesOutShape (Results 1 – 1 of 1) sorted by relevance
587 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput() local626 int32_t* classesOutData, Shape classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat32()650 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat16()672 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant()694 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant()