Searched refs:src_transform_args (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | gen9_wino_convolution.cpp | 397 compute::kernel_arg_list_t src_transform_args; in execute_forward() local
398 src_transform_args.set(0, *src_trans); in execute_forward()
399 src_transform_args.set(1, src); in execute_forward()
403 ctx, src_trans_nd_range, src_trans_kernel_, src_transform_args); in execute_forward()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | gen9_wino_convolution.cpp | 403 compute::kernel_arg_list_t src_transform_args; in execute_forward() local
404 src_transform_args.set(0, *src_trans); in execute_forward()
405 src_transform_args.set(1, src); in execute_forward()
409 ctx, src_trans_nd_range, src_trans_kernel_, src_transform_args); in execute_forward()
|