| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/ |
| H A D | cross_engine_reorder.c | 121 dnnl_exec_arg_t r1_args[] = {{DNNL_ARG_FROM, m_cpu}, {DNNL_ARG_TO, m_gpu}}; in cross_engine_reorder()
124 dnnl_exec_arg_t relu_args[] in cross_engine_reorder()
128 dnnl_exec_arg_t r2_args[] = {{DNNL_ARG_FROM, m_gpu}, {DNNL_ARG_TO, m_cpu}}; in cross_engine_reorder()
|
| H A D | cnn_inference_f32.c | 92 dnnl_exec_arg_t *args;
96 node->args = (dnnl_exec_arg_t *)malloc(sizeof(dnnl_exec_arg_t) * nargs); in prepare_arg_node()
103 static void set_arg(dnnl_exec_arg_t *arg, int arg_idx, dnnl_memory_t memory) { in set_arg()
|
| H A D | cpu_cnn_training_f32.c | 77 dnnl_exec_arg_t *args;
81 node->args = (dnnl_exec_arg_t *)malloc(sizeof(dnnl_exec_arg_t) * nargs); in prepare_arg_node()
88 static void set_arg(dnnl_exec_arg_t *arg, int arg_idx, dnnl_memory_t memory) { in set_arg()
|
| /dports/math/onednn/oneDNN-2.5.1/examples/ |
| H A D | cross_engine_reorder.c | 121 dnnl_exec_arg_t r1_args[] = {{DNNL_ARG_FROM, m_cpu}, {DNNL_ARG_TO, m_gpu}}; in cross_engine_reorder()
124 dnnl_exec_arg_t relu_args[] in cross_engine_reorder()
128 dnnl_exec_arg_t r2_args[] = {{DNNL_ARG_FROM, m_gpu}, {DNNL_ARG_TO, m_cpu}}; in cross_engine_reorder()
|
| H A D | cnn_inference_f32.c | 92 dnnl_exec_arg_t *args;
96 node->args = (dnnl_exec_arg_t *)malloc(sizeof(dnnl_exec_arg_t) * nargs); in prepare_arg_node()
103 static void set_arg(dnnl_exec_arg_t *arg, int arg_idx, dnnl_memory_t memory) { in set_arg()
|
| H A D | cpu_cnn_training_f32.c | 77 dnnl_exec_arg_t *args;
81 node->args = (dnnl_exec_arg_t *)malloc(sizeof(dnnl_exec_arg_t) * nargs); in prepare_arg_node()
88 static void set_arg(dnnl_exec_arg_t *arg, int arg_idx, dnnl_memory_t memory) { in set_arg()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/ |
| H A D | dnnl_common.cpp | 124 const args_t &args, std::vector<dnnl_exec_arg_t> &dnnl_args) { in execute_unmap_args()
143 std::vector<dnnl_exec_arg_t> dnnl_args; in execute_and_wait()
162 const std::vector<dnnl_exec_arg_t> &dnnl_args) { in primitive_executor()
197 perf_function_t &perf_func, std::vector<dnnl_exec_arg_t> &dnnl_args) { in measure_perf_individual()
208 perf_function_t &perf_func, std::vector<dnnl_exec_arg_t> &dnnl_args) { in measure_perf_aggregate()
250 std::vector<dnnl_exec_arg_t> dnnl_args; in measure_perf()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/ |
| H A D | api.c | 209 dnnl_exec_arg_t c3_args[4] = { in test2()
228 dnnl_exec_arg_t r_args[2] = { in test2()
314 dnnl_exec_arg_t l2_args[2] = { in test3()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/ |
| H A D | api.c | 213 dnnl_exec_arg_t c3_args[4] = { in test2()
232 dnnl_exec_arg_t r_args[2] = { in test2()
321 dnnl_exec_arg_t l2_args[2] = { in test3()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/oneapi/dnnl/ |
| H A D | dnnl_sycl.h | 158 const dnnl_exec_arg_t *args, const void *deps, void *return_event);
|
| H A D | dnnl_sycl.hpp | 294 std::vector<dnnl_exec_arg_t> c_args;
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/ |
| H A D | dnnl_sycl.h | 158 const dnnl_exec_arg_t *args, const void *deps, void *return_event);
|
| H A D | dnnl_sycl.hpp | 294 std::vector<dnnl_exec_arg_t> c_args;
|
| /dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/ |
| H A D | dnnl_sycl.h | 158 const dnnl_exec_arg_t *args, const void *deps, void *return_event);
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/ |
| H A D | dnnl_common.cpp | 132 const args_t &args, std::vector<dnnl_exec_arg_t> &dnnl_args) { in execute_unmap_args()
151 std::vector<dnnl_exec_arg_t> dnnl_args; in execute_and_wait()
170 const std::vector<dnnl_exec_arg_t> &dnnl_args) { in primitive_executor()
205 perf_function_t &perf_func, std::vector<dnnl_exec_arg_t> &dnnl_args) { in measure_perf_individual()
216 perf_function_t &perf_func, std::vector<dnnl_exec_arg_t> &dnnl_args) { in measure_perf_aggregate()
260 std::vector<dnnl_exec_arg_t> dnnl_args; in measure_perf()
|
| /dports/math/onednn/oneDNN-2.5.1/src/sycl/capi/ |
| H A D | capi_primitive.cpp | 33 const dnnl_exec_arg_t *args, const void *deps_, void *return_event_) { in dnnl_sycl_interop_primitive_execute()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/sycl/capi/ |
| H A D | capi_primitive.cpp | 33 const dnnl_exec_arg_t *args, const void *deps_, void *return_event_) { in dnnl_sycl_interop_primitive_execute()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | primitive_exec_types.hpp | 61 const dnnl_exec_arg_t *c_args, exec_args_t &args);
|
| H A D | primitive_exec_types.cpp | 28 const dnnl_exec_arg_t *c_args, exec_args_t &args) { in cvt_primitive_args()
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | primitive_exec_types.hpp | 61 const dnnl_exec_arg_t *c_args, exec_args_t &args);
|
| H A D | primitive_exec_types.cpp | 28 const dnnl_exec_arg_t *c_args, exec_args_t &args) { in cvt_primitive_args()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/zeropad/ |
| H A D | zeropad.cpp | 115 const dnnl_stream_t &stream, const std::vector<dnnl_exec_arg_t> &args) { in perf_func()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/zeropad/ |
| H A D | zeropad.cpp | 115 const dnnl_stream_t &stream, const std::vector<dnnl_exec_arg_t> &args) { in perf_func()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/api/ |
| H A D | test_memory_map.cpp | 127 dnnl_exec_arg_t reorder_args[2] in HANDLE_EXCEPTIONS_FOR_TEST_P()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/gtests/api/ |
| H A D | test_memory_map.cpp | 127 dnnl_exec_arg_t reorder_args[2] in HANDLE_EXCEPTIONS_FOR_TEST_P()
|