Home
last modified time | relevance | path

Searched refs:DNN_SAFE_V (Results 1 – 13 of 13) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/
H A Ddnnl_common.cpp100 DNN_SAFE_V(dnnl_set_cpu_isa_hints(dnnl_cpu_isa_no_hints)); in init_isa_settings()
102 DNN_SAFE_V(dnnl_set_cpu_isa_hints(dnnl_cpu_isa_prefer_ymm)); in init_isa_settings()
192 DNN_SAFE_V(dnnl_engine_get_kind(engine, &engine_kind)); in get_engine_kind()
721 DNN_SAFE_V(dnnl_engine_create(&engine_, engine_kind, idx)); in engine_t()
735 DNN_SAFE_V(dnnl_engine_get_kind(other.engine_, &engine_kind)); in engine_t()
743 DNN_SAFE_V(dnnl_sycl_interop_engine_create(&engine_, dev, ctx)); in engine_t()
745 DNN_SAFE_V(dnnl_engine_create(&engine_, dnnl_cpu, 0)); in engine_t()
751 DNN_SAFE_V(dnnl_ocl_interop_get_device(other.engine_, &dev)); in engine_t()
753 DNN_SAFE_V(dnnl_ocl_interop_engine_create(&engine_, dev, ctx)); in engine_t()
767 if (is_owner_) DNN_SAFE_V(dnnl_engine_destroy(engine_)); in ~engine_t()
[all …]
H A Ddnnl_common.hpp55 #define DNN_SAFE_V(f) \ macro
254 DNN_SAFE_V(dnnl_primitive_destroy(t)); in destroy()
261 DNN_SAFE_V(dnnl_primitive_desc_destroy(t)); in destroy()
268 DNN_SAFE_V(dnnl_primitive_attr_destroy(t)); in destroy()
H A Ddnn_types.cpp808 DNN_SAFE_V(dnnl_primitive_attr_create(&dnnl_attr)); in create_dnnl_attr()
818 DNN_SAFE_V(dnnl_primitive_attr_set_output_scales( in create_dnnl_attr()
832 DNN_SAFE_V(dnnl_primitive_attr_set_scales( in create_dnnl_attr()
850 DNN_SAFE_V(dnnl_primitive_attr_set_zero_points( in create_dnnl_attr()
857 DNN_SAFE_V(dnnl_post_ops_create(&ops)); in create_dnnl_attr()
863 DNN_SAFE_V(dnnl_post_ops_append_sum_v2( in create_dnnl_attr()
880 DNN_SAFE_V(dnnl_post_ops_append_dw(ops, wei_dt, bia_dt, in create_dnnl_attr()
889 DNN_SAFE_V(dnnl_post_ops_append_binary( in create_dnnl_attr()
895 DNN_SAFE_V(dnnl_primitive_attr_set_post_ops(dnnl_attr, ops)); in create_dnnl_attr()
901 DNN_SAFE_V(dnnl_post_ops_destroy(ops)); in create_dnnl_attr()
[all …]
H A Ddnnl_memory.hpp191 DNN_SAFE_V(dnnl_memory_map_data(m_, &mapped_ptr_)); in map()
199 DNN_SAFE_V(dnnl_memory_unmap_data(m_, mapped_ptr_)); in unmap()
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/
H A Ddnnl_common.cpp100 DNN_SAFE_V(dnnl_set_cpu_isa_hints(dnnl_cpu_isa_no_hints)); in init_isa_settings()
102 DNN_SAFE_V(dnnl_set_cpu_isa_hints(dnnl_cpu_isa_prefer_ymm)); in init_isa_settings()
200 DNN_SAFE_V(dnnl_engine_get_kind(engine, &engine_kind)); in get_engine_kind()
800 DNN_SAFE_V(dnnl_engine_create(&engine_, engine_kind, idx)); in engine_t()
814 DNN_SAFE_V(dnnl_engine_get_kind(other.engine_, &engine_kind)); in engine_t()
822 DNN_SAFE_V(dnnl_sycl_interop_engine_create(&engine_, dev, ctx)); in engine_t()
824 DNN_SAFE_V(dnnl_engine_create(&engine_, dnnl_cpu, 0)); in engine_t()
830 DNN_SAFE_V(dnnl_ocl_interop_get_device(other.engine_, &dev)); in engine_t()
832 DNN_SAFE_V(dnnl_ocl_interop_engine_create(&engine_, dev, ctx)); in engine_t()
846 if (is_owner_) DNN_SAFE_V(dnnl_engine_destroy(engine_)); in ~engine_t()
[all …]
H A Ddnn_types.cpp883 DNN_SAFE_V(dnnl_primitive_attr_create(&dnnl_attr)); in create_dnnl_attr()
893 DNN_SAFE_V(dnnl_primitive_attr_set_output_scales( in create_dnnl_attr()
907 DNN_SAFE_V(dnnl_primitive_attr_set_scales( in create_dnnl_attr()
925 DNN_SAFE_V(dnnl_primitive_attr_set_zero_points( in create_dnnl_attr()
932 DNN_SAFE_V(dnnl_post_ops_create(&ops)); in create_dnnl_attr()
938 DNN_SAFE_V(dnnl_post_ops_append_sum_v3( in create_dnnl_attr()
964 DNN_SAFE_V(dnnl_post_ops_append_binary( in create_dnnl_attr()
969 DNN_SAFE_V(dnnl_post_ops_append_prelu(ops, mask)); in create_dnnl_attr()
974 DNN_SAFE_V(dnnl_primitive_attr_set_post_ops(dnnl_attr, ops)); in create_dnnl_attr()
980 DNN_SAFE_V(dnnl_post_ops_destroy(ops)); in create_dnnl_attr()
[all …]
H A Ddnnl_common.hpp56 #define DNN_SAFE_V(f) \ macro
259 DNN_SAFE_V(dnnl_primitive_destroy(t)); in destroy()
266 DNN_SAFE_V(dnnl_primitive_desc_destroy(t)); in destroy()
273 DNN_SAFE_V(dnnl_primitive_attr_destroy(t)); in destroy()
H A Ddnnl_memory.hpp191 DNN_SAFE_V(dnnl_memory_map_data(m_, &mapped_ptr_)); in map()
199 DNN_SAFE_V(dnnl_memory_unmap_data(m_, mapped_ptr_)); in unmap()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/
H A Drnn.cpp102 DNN_SAFE_V(dnnl_primitive_attr_create(dnnl_attr)); in create_dnnl_rnn_attr()
105 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_tparams(*dnnl_attr, true, in create_dnnl_rnn_attr()
108 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_weights_qparams( in create_dnnl_rnn_attr()
112 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_weights_projection_qparams( in create_dnnl_rnn_attr()
117 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_data_qparams( in create_dnnl_rnn_attr()
120 DNN_SAFE_V(dnnl_primitive_attr_set_scratchpad_mode( in create_dnnl_rnn_attr()
H A Dlstm_cell.cpp256 DNN_SAFE_V(diff_dst == nullptr ? dnnl_out_of_memory : dnnl_success); in lstm_bwd()
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/
H A Drnn.cpp102 DNN_SAFE_V(dnnl_primitive_attr_create(dnnl_attr)); in create_dnnl_rnn_attr()
105 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_tparams(*dnnl_attr, true, in create_dnnl_rnn_attr()
108 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_weights_qparams( in create_dnnl_rnn_attr()
112 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_weights_projection_qparams( in create_dnnl_rnn_attr()
117 DNN_SAFE_V(dnnl_primitive_attr_set_rnn_data_qparams( in create_dnnl_rnn_attr()
120 DNN_SAFE_V(dnnl_primitive_attr_set_scratchpad_mode( in create_dnnl_rnn_attr()
H A Dlstm_cell.cpp256 DNN_SAFE_V(diff_dst == nullptr ? dnnl_out_of_memory : dnnl_success); in lstm_bwd()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/conv/
H A Dref_conv.cpp30 DNN_SAFE_V(dnnl_primitive_get_primitive_desc(prim_ref, &pd_ref)); in exec_conv()
31 DNN_SAFE_V(dnnl_primitive_desc_query( in exec_conv()