MCPcopy Create free account
hub / github.com/dmlc/xgboost / _array_impl

Function _array_impl

python-package/xgboost/testing/multi_target.py:166–171  ·  view source on GitHub ↗
(device: Device)

Source from the content-addressed store, hash-verified

164
165
166def _array_impl(device: Device) -> ModuleType:
167 if device == "cuda":
168 nda = import_cupy()
169 else:
170 nda = np
171 return nda
172
173
174class LsObj0(TreeObjective):

Callers 5

__call__Method · 0.85
split_gradMethod · 0.85
__call__Method · 0.85
split_gradMethod · 0.85
run_with_iterFunction · 0.85

Calls 1

import_cupyFunction · 0.85

Tested by

no test coverage detected