Searched refs:post_op_reorder (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/ |
| H A D | cudnn_convolution_impl.hpp | 446 post_op_reorder = args[7]; in execute() local
479 execute_reorder(handle, y, post_op_reorder, true); in execute()
480 execute_sum(handle, post_op_reorder, post_op_scratch, in execute()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/ |
| H A D | cudnn_convolution_impl.hpp | 473 post_op_reorder = args[7]; in execute() local
517 execute_reorder(handle, y, post_op_reorder, true); in execute()
518 execute_sum(handle, post_op_reorder, post_op_scratch, in execute()
|