Enqueue the correctly sized batch arrays to Caffe2's queue.
(self, blob_name, queue, data_arr)
| 339 | ) |
| 340 | |
| 341 | def _enqueue(self, blob_name, queue, data_arr): |
| 342 | ''' |
| 343 | Enqueue the correctly sized batch arrays to Caffe2's queue. |
| 344 | ''' |
| 345 | workspace.FeedBlob( |
| 346 | self._scratch_blob[blob_name], |
| 347 | data_arr, |
| 348 | device_option=self._device_option |
| 349 | ) |
| 350 | |
| 351 | op = core.CreateOperator( |
| 352 | "SafeEnqueueBlobs", |
| 353 | [queue, self._scratch_blob[blob_name]], |
| 354 | [self._scratch_blob[blob_name], self._scratch_status[blob_name]], |
| 355 | device_option=self._device_option |
| 356 | ) |
| 357 | workspace.RunOperatorOnce(op) |
| 358 | |
| 359 | def _create_caffe2_queues(self, net): |
| 360 | ''' |
no outgoing calls
no test coverage detected