| /dports/misc/py-gluoncv/gluon-cv-0.9.0/gluoncv/data/ |
| H A D | dataloader.py | 162 pin_memory=False, pin_device_id=0, worker_fn=_worker_fn, prefetch=0): argument
164 worker_pool, batchify_fn, batch_sampler, pin_memory=pin_memory,
260 num_workers=0, pin_memory=False, pin_device_id=0, prefetch=None): argument
264 num_workers=0, pin_memory=pin_memory)
296 self._batch_sampler, pin_memory=self._pin_memory, pin_device_id=self._pin_device_id,
|
| /dports/science/code_saturne/code_saturne-7.1.0/src/alge/ |
| H A D | cs_sles_amgx.c | 462 if (c->pin_memory) { in _setup_matrix_1_ring()
486 if (c->pin_memory) { in _setup_matrix_1_ring()
590 if (c->pin_memory) { in _setup_matrix_dist()
630 if (c->pin_memory) { in _setup_matrix_dist()
680 if (c->pin_memory) { in _setup_update_coeffs()
698 if (c->pin_memory) { in _setup_update_coeffs()
804 c->pin_memory = true; in cs_sles_amgx_create()
846 d->pin_memory = c->pin_memory; in cs_sles_amgx_copy()
1076 c->pin_memory = pin_memory; in cs_sles_amgx_set_pin_memory()
1420 if (c->pin_memory) { in cs_sles_amgx_solve()
[all …]
|
| H A D | cs_sles_amgx.h | 234 bool pin_memory);
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/data/ |
| H A D | dataloader.py | 194 def fetcher_loop_v1(data_queue, data_buffer, pin_memory=False, argument
201 if pin_memory:
215 pin_memory=False, pin_device_id=0, worker_fn=worker_loop_v1): argument
244 args=(self._data_queue, self._data_buffer, pin_memory,
355 num_workers=0, pin_memory=False, pin_device_id=0): argument
357 self._pin_memory = pin_memory
440 def __init__(self, worker_pool, batchify_fn, batch_sampler, pin_memory=False, argument
451 self._pin_memory = pin_memory
622 num_workers=0, pin_memory=False, pin_device_id=0, argument
625 self._pin_memory = pin_memory
[all …]
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/data/ |
| H A D | dataloader.py | 194 def fetcher_loop_v1(data_queue, data_buffer, pin_memory=False, argument
201 if pin_memory:
215 pin_memory=False, pin_device_id=0, worker_fn=worker_loop_v1): argument
244 args=(self._data_queue, self._data_buffer, pin_memory,
355 num_workers=0, pin_memory=False, pin_device_id=0): argument
357 self._pin_memory = pin_memory
440 def __init__(self, worker_pool, batchify_fn, batch_sampler, pin_memory=False, argument
451 self._pin_memory = pin_memory
622 num_workers=0, pin_memory=False, pin_device_id=0, argument
625 self._pin_memory = pin_memory
[all …]
|
| /dports/misc/py-gluonnlp/gluon-nlp-0.10.0/src/gluonnlp/data/ |
| H A D | dataloader.py | 62 def __init__(self, worker_pool, batchify_fn, batch_sampler, pin_memory=False, argument
72 self._pin_memory = pin_memory
174 num_workers=0, pin_memory=False, prefetch=None, thread_pool=False): argument
176 self._pin_memory = pin_memory
235 pin_memory=self._pin_memory,
|
| H A D | datasetloader.py | 78 pin_memory=False, worker_fn=_batch_worker_fn, prefetch=0, argument
87 self._pin_memory = pin_memory
343 pin_memory=False, circle_length=1, argument
385 self._pin_memory = pin_memory
442 pin_memory=self._pin_memory, worker_fn=_batch_worker_fn,
|
| /dports/misc/py-gluonnlp/gluon-nlp-0.10.0/tests/unittest/train/ |
| H A D | test_datasetloader.py | 44 pin_memory=True,
64 pin_memory=True,
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow-ps/ |
| H A D | ps_local-inl.h | 510 bool pin_memory; member
518 if (pin_memory) { in ~PushEntry()
533 bool pin_memory, bool need_weight) { in Init()
534 this->pin_memory = pin_memory; in Init()
537 if (pin_memory) { in Init()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow-ps/ |
| H A D | ps_local-inl.h | 510 bool pin_memory; member
518 if (pin_memory) { in ~PushEntry()
533 bool pin_memory, bool need_weight) { in Init()
534 this->pin_memory = pin_memory; in Init()
537 if (pin_memory) { in Init()
|
| /dports/misc/py-gluoncv/gluon-cv-0.9.0/gluoncv/torch/data/video_cls/ |
| H A D | dataset_classification.py | 280 num_workers=9, pin_memory=True,
285 num_workers=9, sampler=train_sampler, pin_memory=True)
289 num_workers=9, sampler=val_sampler, pin_memory=True)
314 num_workers=9, sampler=test_sampler, pin_memory=True)
|
| /dports/misc/py-gluoncv/gluon-cv-0.9.0/docs/tutorials/depth/ |
| H A D | test_monodepth2_posenet.py | 79 pin_memory=True, last_batch='keep')
|
| /dports/misc/py-gluoncv/gluon-cv-0.9.0/scripts/depth/ |
| H A D | test_pose.py | 73 pin_memory=True, last_batch='keep')
|
| H A D | test.py | 64 pin_memory=True, last_batch='keep')
|
| H A D | trainer.py | 67 pin_memory=True, last_batch='discard')
75 pin_memory=True, last_batch='discard')
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
| H A D | test_gluon_data.py | 294 loader2 = gluon.data.DataLoader(dataset, 8, pin_memory=True)
303 loader3 = gluon.data.DataLoader(dataset, 8, pin_memory=True,
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
| H A D | test_gluon_data.py | 294 loader2 = gluon.data.DataLoader(dataset, 8, pin_memory=True)
303 loader3 = gluon.data.DataLoader(dataset, 8, pin_memory=True,
|
| /dports/misc/py-gluonnlp/gluon-nlp-0.10.0/scripts/bert/ |
| H A D | pretraining_utils.py | 230 pin_memory=False,
308 pin_memory=False,
|
| /dports/math/py-nevergrad/nevergrad-0.4.3.post2/nevergrad/functions/images/ |
| H A D | core.py | 253 ifolder, batch_size=1, shuffle=True, num_workers=8, pin_memory=True
|