Searched refs:pooling_args (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/primitives/ |
| H A D | pooling.cpp | 122 std::unordered_map<int, memory> pooling_args; in pooling_example() local
123 pooling_args.insert({DNNL_ARG_SRC, src_mem}); in pooling_example()
124 pooling_args.insert({DNNL_ARG_DST, dst_mem}); in pooling_example()
125 pooling_args.insert({DNNL_ARG_WORKSPACE, workspace_mem}); in pooling_example()
128 pooling_prim.execute(engine_stream, pooling_args); in pooling_example()
|
| /dports/math/onednn/oneDNN-2.5.1/examples/primitives/ |
| H A D | pooling.cpp | 122 std::unordered_map<int, memory> pooling_args; in pooling_example() local
123 pooling_args.insert({DNNL_ARG_SRC, src_mem}); in pooling_example()
124 pooling_args.insert({DNNL_ARG_DST, dst_mem}); in pooling_example()
125 pooling_args.insert({DNNL_ARG_WORKSPACE, workspace_mem}); in pooling_example()
128 pooling_prim.execute(engine_stream, pooling_args); in pooling_example()
|