| 1085 | } |
| 1086 | |
| 1087 | void CheckAnswersForCocoOnCpu(Workspace *ws, bool offset = false) { |
| 1088 | TensorList<CPUBackend> &boxes = ws->Output<dali::CPUBackend>(0); |
| 1089 | TensorList<CPUBackend> &labels = ws->Output<dali::CPUBackend>(1); |
| 1090 | CheckAnswersForCoco(&boxes, &labels, offset); |
| 1091 | } |
| 1092 | |
| 1093 | void CheckAnswersForCocoOnGpu(Workspace *ws, bool offset = false) { |
| 1094 | auto boxes = this->CopyTensorListToHost(ws->Output<dali::GPUBackend>(0)); |