Home
last modified time | relevance | path

Searched refs:in_md (Results 1 – 18 of 18) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Dmemory.cpp269 if (any_null(out_md, in_md) || !memory_desc_sanity_check(in_md) in dnnl_memory_desc_reshape()
271 ndims, dims, in_md->data_type, in_md->format_kind) in dnnl_memory_desc_reshape()
274 || types::is_zero_md(in_md) in dnnl_memory_desc_reshape()
275 || volume(in_md->dims, in_md->ndims) != volume(dims, ndims) in dnnl_memory_desc_reshape()
277 || in_md->extra.flags != 0) in dnnl_memory_desc_reshape()
285 assert(in_md->extra.flags == 0); in dnnl_memory_desc_reshape()
288 auto md = *in_md; in dnnl_memory_desc_reshape()
293 const int i_ndims = in_md->ndims; in dnnl_memory_desc_reshape()
296 const auto &i_dims = in_md->dims, &i_pdims = in_md->padded_dims; in dnnl_memory_desc_reshape()
458 if (any_null(out_md, in_md) || !memory_desc_sanity_check(in_md) in dnnl_memory_desc_permute_axes()
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/common/
H A Dmemory.cpp269 if (any_null(out_md, in_md) || !memory_desc_sanity_check(in_md) in dnnl_memory_desc_reshape()
271 ndims, dims, in_md->data_type, in_md->format_kind) in dnnl_memory_desc_reshape()
274 || types::is_zero_md(in_md) in dnnl_memory_desc_reshape()
275 || volume(in_md->dims, in_md->ndims) != volume(dims, ndims) in dnnl_memory_desc_reshape()
277 || in_md->extra.flags != 0) in dnnl_memory_desc_reshape()
285 assert(in_md->extra.flags == 0); in dnnl_memory_desc_reshape()
288 auto md = *in_md; in dnnl_memory_desc_reshape()
293 const int i_ndims = in_md->ndims; in dnnl_memory_desc_reshape()
296 const auto &i_dims = in_md->dims, &i_pdims = in_md->padded_dims; in dnnl_memory_desc_reshape()
458 if (any_null(out_md, in_md) || !memory_desc_sanity_check(in_md) in dnnl_memory_desc_permute_axes()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/
H A Dtest_reorder_formats.cpp87 md in_md, out_md; in SetUp() local
123 in_md = md(i_dims, in_dt, in_tag, true); in SetUp()
124 if (in_md) break; in SetUp()
126 ASSERT_TRUE(in_md); in SetUp()
128 const dnnl::impl::memory_desc_wrapper in_d(in_md.data); in SetUp()
146 if (in_md.data.ndims != out_md.data.ndims) continue; in SetUp()
163 [=]() { TestFormat(in_md, out_md); }, false, in SetUp()
171 void TestFormat(const md &in_md, const md &out_md) { in TestFormat() argument
172 auto src = test::make_memory(in_md, e); in TestFormat()
175 e, in_md, e, out_md, primitive_attr(), true); in TestFormat()
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/
H A Dtest_reorder_formats.cpp87 md in_md, out_md; in SetUp() local
123 in_md = md(i_dims, in_dt, in_tag, true); in SetUp()
124 if (in_md) break; in SetUp()
126 ASSERT_TRUE(in_md); in SetUp()
128 const dnnl::impl::memory_desc_wrapper in_d(in_md.data); in SetUp()
146 if (in_md.data.ndims != out_md.data.ndims) continue; in SetUp()
163 [=]() { TestFormat(in_md, out_md); }, false, in SetUp()
171 void TestFormat(const md &in_md, const md &out_md) const { in TestFormat() argument
172 auto src = test::make_memory(in_md, e); in TestFormat()
175 e, in_md, e, out_md, primitive_attr(), true); in TestFormat()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/
H A Dgpu_inner_product_pd.hpp81 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t &in_md) { in template_set_default_params() argument
83 if (memory_desc_matches_one_of_tag(in_md, ba, cba, cdba, cdeba)) in template_set_default_params()
85 else if (memory_desc_matches_one_of_tag(in_md, acb, acdb, acdeb)) in template_set_default_params()
88 memory_desc_wrapper md_desc_wrapper(in_md); in template_set_default_params()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/
H A Dgpu_inner_product_pd.hpp81 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t &in_md) { in template_set_default_params() argument
83 if (memory_desc_matches_one_of_tag(in_md, ba, cba, cdba, cdeba)) in template_set_default_params()
85 else if (memory_desc_matches_one_of_tag(in_md, acb, acdb, acdeb)) in template_set_default_params()
88 memory_desc_wrapper md_desc_wrapper(in_md); in template_set_default_params()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/
H A Dconvolution_inner_product.cpp56 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t *in_md) { in init_conf() argument
57 max_dims = adjust_dims(dims, in_md, conf.ndims); in init_conf()
58 if (in_md->format_kind == format_kind::any) { in init_conf()
60 &out_md, max_dims, dims, in_md->data_type, format_tag::any); in init_conf()
62 out_md = *in_md; in init_conf()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/
H A Dconvolution_inner_product.cpp56 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t *in_md) { in init_conf() argument
57 max_dims = adjust_dims(dims, in_md, conf.ndims); in init_conf()
58 if (in_md->format_kind == format_kind::any) { in init_conf()
60 &out_md, max_dims, dims, in_md->data_type, format_tag::any); in init_conf()
62 out_md = *in_md; in init_conf()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/api/
H A Dtest_memory_desc_ops.cpp120 void Test(const memory::desc &in_md, const memory::desc &out_md) { in Test() argument
121 memory::desc get_out_md = in_md.reshape(out_md.dims()); in Test()
123 debug::print_md("in_md", in_md); in Test()
225 void Test(const memory::desc &in_md, const memory::desc &out_md, in Test() argument
227 memory::desc get_out_md = in_md.permute_axes(perm); in Test()
229 debug::print_md("in_md", in_md); in Test()
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/api/
H A Dtest_memory_desc_ops.cpp120 void Test(const memory::desc &in_md, const memory::desc &out_md) { in Test() argument
121 memory::desc get_out_md = in_md.reshape(out_md.dims()); in Test()
123 debug::print_md("in_md", in_md); in Test()
225 void Test(const memory::desc &in_md, const memory::desc &out_md, in Test() argument
227 memory::desc get_out_md = in_md.permute_axes(perm); in Test()
229 debug::print_md("in_md", in_md); in Test()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/
H A Dcudnn_gemm_inner_product.hpp126 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t &in_md) { in template_set_default_params() argument
128 if (memory_desc_matches_one_of_tag(in_md, ab, abc, abcd, abcde)) in template_set_default_params()
130 else if (memory_desc_matches_one_of_tag(in_md, acb, acdb, acdeb)) in template_set_default_params()
132 else if (memory_desc_matches_one_of_tag(in_md, ba, cba, cdba, cdeba)) in template_set_default_params()
135 memory_desc_wrapper md_desc_wrapper(in_md); in template_set_default_params()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/
H A Dcudnn_gemm_inner_product.hpp126 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t &in_md) { in template_set_default_params() argument
128 if (memory_desc_matches_one_of_tag(in_md, ab, abc, abcd, abcde)) in template_set_default_params()
130 else if (memory_desc_matches_one_of_tag(in_md, acb, acdb, acdeb)) in template_set_default_params()
132 else if (memory_desc_matches_one_of_tag(in_md, ba, cba, cdba, cdeba)) in template_set_default_params()
135 memory_desc_wrapper md_desc_wrapper(in_md); in template_set_default_params()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/
H A Dmkldnn_slice.cc53 auto in_md = in.GetMKLDNNData()->get_desc(); in MKLDNNSliceFwd() local
55 auto sub_md = in_md.submemory_desc(dims, offsets); in MKLDNNSliceFwd()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/
H A Dmkldnn_slice.cc53 auto in_md = in.GetMKLDNNData()->get_desc(); in MKLDNNSliceFwd() local
55 auto sub_md = in_md.submemory_desc(dims, offsets); in MKLDNNSliceFwd()
/dports/x11-wm/enlightenment/enlightenment-0.24.2/src/bin/
H A De_slider.c487 static int in_md = 0; in _e_smart_event_mouse_down() local
490 if (in_md > 0) return; in _e_smart_event_mouse_down()
491 in_md++; in _e_smart_event_mouse_down()
514 in_md--; in _e_smart_event_mouse_down()
/dports/net/samba412/samba-4.12.15/source4/heimdal/lib/krb5/
H A Dinit_creds_pw.c1151 METHOD_DATA *in_md, in process_pa_data_to_md() argument
1168 _krb5_debug(context, 5, "KDC send %d patypes", in_md->len); in process_pa_data_to_md()
1169 for (i = 0; i < in_md->len; i++) in process_pa_data_to_md()
1170 _krb5_debug(context, 5, "KDC send PA-DATA type: %d", in_md->val[i].padata_type); in process_pa_data_to_md()
1201 } else if (in_md->len != 0) { in process_pa_data_to_md()
1208 ppaid = process_pa_info(context, creds->client, a, paid, in_md); in process_pa_data_to_md()
/dports/net/py-libcloud/apache-libcloud-3.4.1/test/compute/
H A Dtest_gce.py1365 in_md = [{'k0': 'v0'}, {'k1': 'v1'}]
1374 in_md = {'key': 'k0', 'value': 'v0'}
1383 in_md = {'k0': 'v0'}
1392 in_md = {'k0': 'v0', 'k1': 'v1', 'k2': 'v2'}
1405 in_md = {'items': [{'key': 'k0',
1416 in_md = {'items': 'foo'}
1418 in_md)
1419 in_md = {'items': {'key': 'k1', 'value': 'v0'}}
1421 in_md)
1422 in_md = ['k0', 'v1']
[all …]
/dports/net-mgmt/wmi-client/wmi-1.3.16/Samba/source/heimdal/lib/krb5/
H A Dinit_creds_pw.c1080 METHOD_DATA *in_md, in process_pa_data_to_md() argument
1095 if (in_md->len != 0) { in process_pa_data_to_md()
1101 ppaid = process_pa_info(context, creds->client, a, &paid, in_md); in process_pa_data_to_md()