Searched refs:dst_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 | 414 compute::kernel_arg_list_t dst_transform_args; in execute_forward() local
415 dst_transform_args.set(0, dst); in execute_forward()
416 dst_transform_args.set(1, *M_buf); in execute_forward()
417 dst_transform_args.set(2, bias); in execute_forward()
419 ctx, dst_transform_args, 3, attr_info.all_post_ops); in execute_forward()
423 ctx, dst_trans_nd_range, dst_trans_kernel_, dst_transform_args); in execute_forward()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | gen9_wino_convolution.cpp | 420 compute::kernel_arg_list_t dst_transform_args; in execute_forward() local
421 dst_transform_args.set(0, dst); in execute_forward()
422 dst_transform_args.set(1, *M_buf); in execute_forward()
423 dst_transform_args.set(2, bias); in execute_forward()
425 ctx, dst_transform_args, 3, pd()->attr()->post_ops_); in execute_forward()
429 ctx, dst_trans_nd_range, dst_trans_kernel_, dst_transform_args); in execute_forward()
|