Home
last modified time | relevance | path

Searched refs:dst_mem (Results 1 – 25 of 154) sorted by relevance

1234567

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/
H A Dmemory_format_propagation.cpp185 auto dst_mem = memory( in memory_format_propagation_tutorial() local
218 bool need_reorder_dst = conv_pd.dst_desc() != dst_mem.get_desc(); in memory_format_propagation_tutorial()
241 = need_reorder_dst ? memory(pool_pd.dst_desc(), eng) : dst_mem; in memory_format_propagation_tutorial()
304 auto reorder_dst = reorder(pool_dst_mem, dst_mem); in memory_format_propagation_tutorial()
306 s, {{DNNL_ARG_FROM, pool_dst_mem}, {DNNL_ARG_TO, dst_mem}}); in memory_format_propagation_tutorial()
H A Dgetting_started.cpp273 auto dst_mem = memory(src_md, eng); in getting_started_tutorial() local
391 {DNNL_ARG_DST, dst_mem}, // Destination tag and memory obj in getting_started_tutorial()
434 read_from_dnnl_memory(relu_image.data(), dst_mem); in getting_started_tutorial()
/dports/math/onednn/oneDNN-2.5.1/examples/
H A Dmemory_format_propagation.cpp185 auto dst_mem = memory( in memory_format_propagation_tutorial() local
218 bool need_reorder_dst = conv_pd.dst_desc() != dst_mem.get_desc(); in memory_format_propagation_tutorial()
241 = need_reorder_dst ? memory(pool_pd.dst_desc(), eng) : dst_mem; in memory_format_propagation_tutorial()
304 auto reorder_dst = reorder(pool_dst_mem, dst_mem); in memory_format_propagation_tutorial()
306 s, {{DNNL_ARG_FROM, pool_dst_mem}, {DNNL_ARG_TO, dst_mem}}); in memory_format_propagation_tutorial()
H A Dgetting_started.cpp273 auto dst_mem = memory(src_md, eng); in getting_started_tutorial() local
391 {DNNL_ARG_DST, dst_mem}, // Destination tag and memory obj in getting_started_tutorial()
434 read_from_dnnl_memory(relu_image.data(), dst_mem); in getting_started_tutorial()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/mkldnn_compat/examples/
H A Dcpu_memory_format_propagation.cpp167 auto dst_mem = memory( in cpu_memory_format_propagation_tutorial() local
200 bool need_reorder_dst = conv_pd.dst_desc() != dst_mem.get_desc(); in cpu_memory_format_propagation_tutorial()
225 : dst_mem; in cpu_memory_format_propagation_tutorial()
289 auto reorder_dst = reorder(pool_dst_mem, dst_mem); in cpu_memory_format_propagation_tutorial()
291 {{MKLDNN_ARG_FROM, pool_dst_mem}, {MKLDNN_ARG_TO, dst_mem}}); in cpu_memory_format_propagation_tutorial()
H A Dgetting_started.cpp273 auto dst_mem = memory(src_md, eng); in getting_started_tutorial() local
391 {MKLDNN_ARG_DST, dst_mem}, // Destination tag and memory obj in getting_started_tutorial()
434 read_from_mkldnn_memory(relu_image.data(), dst_mem); in getting_started_tutorial()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/primitives/
H A Dshuffle.cpp82 auto dst_mem = memory({src_dims, dt::f32, tag::abcd}, engine); in shuffle_example() local
100 shuffle_args.insert({DNNL_ARG_DST, dst_mem}); in shuffle_example()
109 read_from_dnnl_memory(dst_data.data(), dst_mem); in shuffle_example()
H A Dreduction.cpp73 auto dst_mem = memory(dst_md, engine); in reduction_example() local
91 reduction_args.insert({DNNL_ARG_DST, dst_mem}); in reduction_example()
100 read_from_dnnl_memory(dst_data.data(), dst_mem); in reduction_example()
H A Deltwise.cpp80 auto dst_mem = memory(dst_md, engine); in eltwise_example() local
98 eltwise_args.insert({DNNL_ARG_DST, dst_mem}); in eltwise_example()
107 read_from_dnnl_memory(dst_data.data(), dst_mem); in eltwise_example()
H A Dlrn.cpp74 auto dst_mem = memory(src_md, engine); in lrn_example() local
103 lrn_args.insert({DNNL_ARG_DST, dst_mem}); in lrn_example()
112 read_from_dnnl_memory(dst_data.data(), dst_mem); in lrn_example()
H A Dreorder.cpp79 auto dst_mem = memory(dst_md, engine); in reorder_example() local
108 reorder_args.insert({DNNL_ARG_DST, dst_mem}); in reorder_example()
117 read_from_dnnl_memory(dst_data.data(), dst_mem); in reorder_example()
H A Dresampling.cpp80 auto dst_mem = memory(dst_md, engine); in resampling_example() local
99 resampling_args.insert({DNNL_ARG_DST, dst_mem}); in resampling_example()
108 read_from_dnnl_memory(dst_data.data(), dst_mem); in resampling_example()
H A Dsum.cpp104 auto dst_mem = memory(sum_pd.dst_desc(), engine); in sum_example() local
108 sum_args.insert({DNNL_ARG_DST, dst_mem}); in sum_example()
120 read_from_dnnl_memory(dst_data.data(), dst_mem); in sum_example()
H A Dpooling.cpp99 auto dst_mem = memory(dst_md, engine); in pooling_example() local
124 pooling_args.insert({DNNL_ARG_DST, dst_mem}); in pooling_example()
134 read_from_dnnl_memory(dst_data.data(), dst_mem); in pooling_example()
H A Dmatmul.cpp95 auto dst_mem = memory(dst_md, engine); in matmul_example() local
125 matmul_args.insert({DNNL_ARG_DST, dst_mem}); in matmul_example()
134 read_from_dnnl_memory(dst_data.data(), dst_mem); in matmul_example()
H A Dinner_product.cpp98 auto dst_mem = memory(dst_md, engine); in inner_product_example() local
159 inner_product_args.insert({DNNL_ARG_DST, dst_mem}); in inner_product_example()
168 read_from_dnnl_memory(dst_data.data(), dst_mem); in inner_product_example()
/dports/math/onednn/oneDNN-2.5.1/examples/primitives/
H A Dshuffle.cpp82 auto dst_mem = memory({src_dims, dt::f32, tag::abcd}, engine); in shuffle_example() local
100 shuffle_args.insert({DNNL_ARG_DST, dst_mem}); in shuffle_example()
109 read_from_dnnl_memory(dst_data.data(), dst_mem); in shuffle_example()
H A Deltwise.cpp80 auto dst_mem = memory(dst_md, engine); in eltwise_example() local
98 eltwise_args.insert({DNNL_ARG_DST, dst_mem}); in eltwise_example()
107 read_from_dnnl_memory(dst_data.data(), dst_mem); in eltwise_example()
H A Dreduction.cpp73 auto dst_mem = memory(dst_md, engine); in reduction_example() local
91 reduction_args.insert({DNNL_ARG_DST, dst_mem}); in reduction_example()
100 read_from_dnnl_memory(dst_data.data(), dst_mem); in reduction_example()
H A Dlrn.cpp74 auto dst_mem = memory(src_md, engine); in lrn_example() local
103 lrn_args.insert({DNNL_ARG_DST, dst_mem}); in lrn_example()
112 read_from_dnnl_memory(dst_data.data(), dst_mem); in lrn_example()
H A Dresampling.cpp80 auto dst_mem = memory(dst_md, engine); in resampling_example() local
99 resampling_args.insert({DNNL_ARG_DST, dst_mem}); in resampling_example()
108 read_from_dnnl_memory(dst_data.data(), dst_mem); in resampling_example()
H A Dreorder.cpp79 auto dst_mem = memory(dst_md, engine); in reorder_example() local
108 reorder_args.insert({DNNL_ARG_DST, dst_mem}); in reorder_example()
117 read_from_dnnl_memory(dst_data.data(), dst_mem); in reorder_example()
H A Dsum.cpp104 auto dst_mem = memory(sum_pd.dst_desc(), engine); in sum_example() local
108 sum_args.insert({DNNL_ARG_DST, dst_mem}); in sum_example()
120 read_from_dnnl_memory(dst_data.data(), dst_mem); in sum_example()
H A Dpooling.cpp99 auto dst_mem = memory(dst_md, engine); in pooling_example() local
124 pooling_args.insert({DNNL_ARG_DST, dst_mem}); in pooling_example()
134 read_from_dnnl_memory(dst_data.data(), dst_mem); in pooling_example()
H A Dmatmul.cpp95 auto dst_mem = memory(dst_md, engine); in matmul_example() local
125 matmul_args.insert({DNNL_ARG_DST, dst_mem}); in matmul_example()
134 read_from_dnnl_memory(dst_data.data(), dst_mem); in matmul_example()

1234567