Lines Matching refs:getInputShape

231     Shape roiShape = context->getInputShape(kRoiTensor);  in prepare()
232 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
233 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
234 Shape imageInfoShape = context->getInputShape(kImageInfoTensor); in prepare()
280 context->getInputShape(kRoiTensor), in execute()
282 context->getInputShape(kDeltaTensor), in execute()
284 context->getInputShape(kBatchesTensor), in execute()
286 context->getInputShape(kImageInfoTensor), in execute()
292 context->getInputShape(kRoiTensor), in execute()
294 context->getInputShape(kDeltaTensor), in execute()
296 context->getInputShape(kBatchesTensor), in execute()
298 context->getInputShape(kImageInfoTensor), in execute()
305 context->getInputShape(kRoiTensor), in execute()
307 context->getInputShape(kDeltaTensor), in execute()
309 context->getInputShape(kBatchesTensor), in execute()
311 context->getInputShape(kImageInfoTensor), in execute()
316 context->getInputShape(kRoiTensor), in execute()
318 context->getInputShape(kDeltaTensor), in execute()
320 context->getInputShape(kBatchesTensor), in execute()
322 context->getInputShape(kImageInfoTensor), in execute()
573 Shape scoresShape = context->getInputShape(kScoreTensor); in boxWithNmsLimitWriteOutput()
760 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
761 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
762 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare()
816 if (getSizeOfDimension(context->getInputShape(kScoreTensor), 0) == 0) return true; in execute()
821 context->getInputShape(kScoreTensor), in execute()
823 context->getInputShape(kRoiTensor), in execute()
825 context->getInputShape(kBatchesTensor), in execute()
843 context->getInputShape(kScoreTensor), in execute()
845 context->getInputShape(kRoiTensor), in execute()
847 context->getInputShape(kBatchesTensor), in execute()
865 context->getInputShape(kScoreTensor), in execute()
867 context->getInputShape(kRoiTensor), in execute()
869 context->getInputShape(kBatchesTensor), in execute()
887 context->getInputShape(kScoreTensor), in execute()
889 context->getInputShape(kRoiTensor), in execute()
891 context->getInputShape(kBatchesTensor), in execute()
1291 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
1292 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare()
1293 Shape anchorsShape = context->getInputShape(kAnchorTensor); in prepare()
1294 Shape imageInfoDataShape = context->getInputShape(kImageInfoTensor); in prepare()
1348 context->getInputShape(kScoreTensor), in execute()
1350 context->getInputShape(kDeltaTensor), in execute()
1352 context->getInputShape(kAnchorTensor), in execute()
1354 context->getInputShape(kImageInfoTensor), in execute()
1365 context->getInputShape(kScoreTensor), in execute()
1367 context->getInputShape(kDeltaTensor), in execute()
1369 context->getInputShape(kAnchorTensor), in execute()
1371 context->getInputShape(kImageInfoTensor), in execute()
1382 context->getInputShape(kScoreTensor), in execute()
1384 context->getInputShape(kDeltaTensor), in execute()
1386 context->getInputShape(kAnchorTensor), in execute()
1388 context->getInputShape(kImageInfoTensor), in execute()
1399 context->getInputShape(kScoreTensor), in execute()
1401 context->getInputShape(kDeltaTensor), in execute()
1403 context->getInputShape(kAnchorTensor), in execute()
1405 context->getInputShape(kImageInfoTensor), in execute()
1620 Shape scoreShape = context->getInputShape(kScoreTensor); in prepare()
1621 Shape deltasShape = context->getInputShape(kDeltaTensor); in prepare()
1622 Shape anchorsShape = context->getInputShape(kAnchorTensor); in prepare()
1698 context->getInputShape(kScoreTensor), in execute()
1700 context->getInputShape(kDeltaTensor), in execute()
1702 context->getInputShape(kAnchorTensor), in execute()
1726 context->getInputShape(kScoreTensor), in execute()
1728 context->getInputShape(kDeltaTensor), in execute()
1730 context->getInputShape(kAnchorTensor), in execute()