Home
last modified time | relevance | path

Searched refs:ctx_dev (Results 1 – 24 of 24) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/cpp-package/example/
H A Dmlp.cpp75 Context ctx_dev(DeviceType::kCPU, 0); in MLP() local
77 NDArray array_x(Shape(128, 28), ctx_dev, false); in MLP()
78 NDArray array_y(Shape(128), ctx_dev, false); in MLP()
96 NDArray array_w_1(Shape(512, 28), ctx_dev, false); in MLP()
97 NDArray array_b_1(Shape(512), ctx_dev, false); in MLP()
98 NDArray array_w_2(Shape(10, 512), ctx_dev, false); in MLP()
99 NDArray array_b_2(Shape(10), ctx_dev, false); in MLP()
110 NDArray array_w_1_g(Shape(512, 28), ctx_dev, false); in MLP()
111 NDArray array_b_1_g(Shape(512), ctx_dev, false); in MLP()
112 NDArray array_w_2_g(Shape(10, 512), ctx_dev, false); in MLP()
[all …]
H A Dlenet.cpp37 ctx_dev(Context(DeviceType::kCPU, 0)) in Lenet()
39 ctx_dev(Context(DeviceType::kGPU, 0)) in Lenet()
122 args_map["data"] = data_array.Slice(0, batch_size).Copy(ctx_dev); in Run()
123 args_map["data_label"] = label_array.Slice(0, batch_size).Copy(ctx_dev); in Run()
138 lenet.InferArgsMap(ctx_dev, &args_map, args_map); in Run()
146 Executor *exe = lenet.SimpleBind(ctx_dev, args_map); in Run()
157 .Copy(ctx_dev); in Run()
160 .Copy(ctx_dev); in Run()
182 Context ctx_dev; member in Lenet
221 val_data.Slice(start_index, start_index + batch_size).Copy(ctx_dev); in ValAccuracy()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/cpp-package/example/
H A Dmlp.cpp75 Context ctx_dev(DeviceType::kCPU, 0); in MLP() local
77 NDArray array_x(Shape(128, 28), ctx_dev, false); in MLP()
78 NDArray array_y(Shape(128), ctx_dev, false); in MLP()
96 NDArray array_w_1(Shape(512, 28), ctx_dev, false); in MLP()
97 NDArray array_b_1(Shape(512), ctx_dev, false); in MLP()
98 NDArray array_w_2(Shape(10, 512), ctx_dev, false); in MLP()
99 NDArray array_b_2(Shape(10), ctx_dev, false); in MLP()
110 NDArray array_w_1_g(Shape(512, 28), ctx_dev, false); in MLP()
111 NDArray array_b_1_g(Shape(512), ctx_dev, false); in MLP()
112 NDArray array_w_2_g(Shape(10, 512), ctx_dev, false); in MLP()
[all …]
H A Dlenet.cpp37 ctx_dev(Context(DeviceType::kCPU, 0)) in Lenet()
39 ctx_dev(Context(DeviceType::kGPU, 0)) in Lenet()
122 args_map["data"] = data_array.Slice(0, batch_size).Copy(ctx_dev); in Run()
123 args_map["data_label"] = label_array.Slice(0, batch_size).Copy(ctx_dev); in Run()
138 lenet.InferArgsMap(ctx_dev, &args_map, args_map); in Run()
146 Executor *exe = lenet.SimpleBind(ctx_dev, args_map); in Run()
157 .Copy(ctx_dev); in Run()
160 .Copy(ctx_dev); in Run()
182 Context ctx_dev; member in Lenet
221 val_data.Slice(start_index, start_index + batch_size).Copy(ctx_dev); in ValAccuracy()
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/sycl/
H A Dsycl_engine.cpp65 [&](const ::sycl::device &ctx_dev) { in engine_create() argument
66 return are_equal(ctx_dev, dev); in engine_create()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/sycl/
H A Dsycl_engine.cpp65 [&](const cl::sycl::device &ctx_dev) { in engine_create() argument
66 return are_equal(ctx_dev, dev); in engine_create()
/dports/lang/gcc6-aux/gcc-6-20180516/libgomp/plugin/
H A Dplugin-nvptx.c572 CUdevice dev, ctx_dev; in nvptx_open_device() local
586 r = cuCtxGetDevice (&ctx_dev); in nvptx_open_device()
590 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gnat_util/gcc-6-20180516/libgomp/plugin/
H A Dplugin-nvptx.c572 CUdevice dev, ctx_dev; in nvptx_open_device() local
586 r = cuCtxGetDevice (&ctx_dev); in nvptx_open_device()
590 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc10/gcc-10.3.0/libgomp/plugin/
H A Dplugin-nvptx.c402 CUdevice dev, ctx_dev; in nvptx_open_device() local
414 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
421 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/devel/avr-gcc/gcc-10.2.0/libgomp/plugin/
H A Dplugin-nvptx.c402 CUdevice dev, ctx_dev; in nvptx_open_device() local
414 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
421 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc10-devel/gcc-10-20211008/libgomp/plugin/
H A Dplugin-nvptx.c402 CUdevice dev, ctx_dev; in nvptx_open_device() local
414 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
421 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc12-devel/gcc-12-20211205/libgomp/plugin/
H A Dplugin-nvptx.c420 CUdevice dev, ctx_dev; in nvptx_open_device() local
432 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
439 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc8/gcc-8.5.0/libgomp/plugin/
H A Dplugin-nvptx.c714 CUdevice dev, ctx_dev; in nvptx_open_device() local
726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/devel/riscv64-none-elf-gcc/gcc-8.4.0/libgomp/plugin/
H A Dplugin-nvptx.c714 CUdevice dev, ctx_dev; in nvptx_open_device() local
726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc11-devel/gcc-11-20211009/libgomp/plugin/
H A Dplugin-nvptx.c419 CUdevice dev, ctx_dev; in nvptx_open_device() local
431 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
438 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/devel/arm-none-eabi-gcc/gcc-8.4.0/libgomp/plugin/
H A Dplugin-nvptx.c714 CUdevice dev, ctx_dev; in nvptx_open_device() local
726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc9-devel/gcc-9-20211007/libgomp/plugin/
H A Dplugin-nvptx.c741 CUdevice dev, ctx_dev; in nvptx_open_device() local
753 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
760 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/devel/riscv32-unknown-elf-gcc/gcc-8.4.0/libgomp/plugin/
H A Dplugin-nvptx.c714 CUdevice dev, ctx_dev; in nvptx_open_device() local
726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/devel/aarch64-none-elf-gcc/gcc-8.4.0/libgomp/plugin/
H A Dplugin-nvptx.c714 CUdevice dev, ctx_dev; in nvptx_open_device() local
726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/misc/cxx_atomics_pic/gcc-11.2.0/libgomp/plugin/
H A Dplugin-nvptx.c419 CUdevice dev, ctx_dev; in nvptx_open_device() local
431 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
438 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/devel/riscv64-gcc/gcc-8.3.0/libgomp/plugin/
H A Dplugin-nvptx.c714 CUdevice dev, ctx_dev; in nvptx_open_device() local
726 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
733 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc11/gcc-11.2.0/libgomp/plugin/
H A Dplugin-nvptx.c419 CUdevice dev, ctx_dev; in nvptx_open_device() local
431 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
438 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc9/gcc-9.4.0/libgomp/plugin/
H A Dplugin-nvptx.c741 CUdevice dev, ctx_dev; in nvptx_open_device() local
753 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
760 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()
/dports/lang/gcc9-aux/gcc-9.1.0/libgomp/plugin/
H A Dplugin-nvptx.c741 CUdevice dev, ctx_dev; in nvptx_open_device() local
753 r = CUDA_CALL_NOCHECK (cuCtxGetDevice, &ctx_dev); in nvptx_open_device()
760 if (r != CUDA_ERROR_INVALID_CONTEXT && ctx_dev != dev) in nvptx_open_device()