oneDNN Wrapper
|
This is the complete list of members for MaxPool2D, including all inherited members.
arg_dst (defined in MaxPool2D) | MaxPool2D | |
arg_src (defined in MaxPool2D) | MaxPool2D | |
arg_workspace (defined in MaxPool2D) | MaxPool2D | |
MaxPool2D(int kernel_size, int stride_length, dnnl::memory input, std::vector< dnnl::primitive > &net, std::vector< std::unordered_map< int, dnnl::memory >> &net_args, dnnl::engine eng) | MaxPool2D | |
pooling_fwd_pd (defined in MaxPool2D) | MaxPool2D |