| /dports/security/liboqs/liboqs-0.7.0/scripts/ |
| H A D | update_docs_from_yaml.py | 35 out_md.write('# {}\n\n'.format(kem_yaml['name']))
46 out_md.write(', which takes it from:\n')
51 out_md.write('\n')
55 out_md.write('\n## Parameter set summary\n\n')
72 out_md.write('\n')
114 out_md.write('\n')
121 out_md.write('\n## Explanation of Terms\n\n')
145 out_md.write(', which takes it from:\n')
150 out_md.write('\n')
169 out_md.write('\n')
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
| H A D | test_reorder_formats.cpp | 87 md in_md, out_md; in SetUp() local
142 out_md = md(i_dims, out_dt, out_tag, true); in SetUp()
143 if (out_md) break; in SetUp()
145 ASSERT_TRUE(out_md); in SetUp()
146 if (in_md.data.ndims != out_md.data.ndims) continue; in SetUp()
149 out_md.data); in SetUp()
160 out_md.data.extra = i_extra; 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
173 auto dst = test::make_memory(out_md, e); in TestFormat()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
| H A D | test_reorder_formats.cpp | 87 md in_md, out_md; in SetUp() local
142 out_md = md(i_dims, out_dt, out_tag, true); in SetUp()
143 if (out_md) break; in SetUp()
145 ASSERT_TRUE(out_md); in SetUp()
146 if (in_md.data.ndims != out_md.data.ndims) continue; in SetUp()
149 out_md.data); in SetUp()
160 out_md.data.extra = i_extra; 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()
173 auto dst = test::make_memory(out_md, e); in TestFormat()
[all …]
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/ |
| H A D | mkldnn_fully_connected.cc | 37 const mkldnn::memory::desc &out_md) { in GetFCFwdImpl() argument
81 data_md, weight_md, bias_md, out_md); in GetFCFwdImpl()
85 data_md, weight_md, out_md); in GetFCFwdImpl()
95 auto out_md = GetMemDesc(output); in GetFCBwdData() local
106 auto out_md = GetMemDesc(output); in GetFCBwdWeights() local
111 weight_md, bias_md, out_md); in GetFCBwdWeights()
116 weight_md, out_md); in GetFCBwdWeights()
125 const NDArray *bias, const mkldnn::memory::desc &out_md) { in GetFCFwd() argument
154 mkldnn::memory::desc *out_md) { in MKLDNNFCFlattenData() argument
228 &data, &out_md); in MKLDNNFCForward()
[all …]
|
| H A D | mkldnn_fully_connected-inl.h | 80 const mkldnn::memory::desc &out_md);
89 const mkldnn::memory::desc &out_md) in MKLDNNFullyConnectedForward() argument
90 : fwd_pd(GetFCFwdImpl(full_param, is_train, data, weight, bias, out_md)) { in MKLDNNFullyConnectedForward()
107 const NDArray *bias, const mkldnn::memory::desc &out_md);
112 mkldnn::memory::desc *out_md);
|
| H A D | mkldnn_slice.cc | 54 auto out_md = out.GetMKLDNNData()->get_desc(); in MKLDNNSliceFwd() local
59 this->out_ = std::make_shared<mkldnn::memory>(out_md, engine, nullptr); in MKLDNNSliceFwd()
98 auto out_md = out.GetMKLDNNData()->get_desc(); in MKLDNNSlice() local
99 auto out_mem = CreateMKLDNNMem(out, out_md, req); in MKLDNNSlice()
|
| H A D | mkldnn_convolution.cc | 59 auto out_md = GetMemDesc(output); in GetConvFwdImpl() local
146 weight_md, out_md, strides, padding, padding); in GetConvFwdImpl()
150 weight_md, *bias_md_ptr, out_md, strides, padding, in GetConvFwdImpl()
170 weight_md, out_md, strides, dilates, padding, padding); in GetConvFwdImpl()
174 weight_md, *bias_md_ptr, out_md, strides, dilates, in GetConvFwdImpl()
186 auto out_md = GetMemDesc(output); in GetConvBwdData() local
244 weight_md, out_md, strides, padding, padding); in GetConvBwdData()
262 weight_md, out_md, strides, dilates, padding, in GetConvBwdData()
273 auto out_md = GetMemDesc(output); in GetConvBwdWeights() local
331 weight_md, out_md, strides, padding, padding); in GetConvBwdWeights()
[all …]
|
| H A D | mkldnn_deconvolution.cc | 322 out_md(GetMemDesc(out)), in DeconvDescCreator()
344 } else if (out_md.data.format_kind == dnnl_format_kind_any && in ImposePlainWherePadding()
345 out_size != GetMemDescSize(out_md)) { in ImposePlainWherePadding()
346 out_md = GetDesc(out_md, GetDefaultFormat(out_md)); in ImposePlainWherePadding()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/mkldnn/ |
| H A D | mkldnn_fully_connected.cc | 37 const mkldnn::memory::desc &out_md) { in GetFCFwdImpl() argument
81 data_md, weight_md, bias_md, out_md); in GetFCFwdImpl()
85 data_md, weight_md, out_md); in GetFCFwdImpl()
95 auto out_md = GetMemDesc(output); in GetFCBwdData() local
106 auto out_md = GetMemDesc(output); in GetFCBwdWeights() local
111 weight_md, bias_md, out_md); in GetFCBwdWeights()
116 weight_md, out_md); in GetFCBwdWeights()
125 const NDArray *bias, const mkldnn::memory::desc &out_md) { in GetFCFwd() argument
154 mkldnn::memory::desc *out_md) { in MKLDNNFCFlattenData() argument
228 &data, &out_md); in MKLDNNFCForward()
[all …]
|
| H A D | mkldnn_fully_connected-inl.h | 80 const mkldnn::memory::desc &out_md);
89 const mkldnn::memory::desc &out_md) in MKLDNNFullyConnectedForward() argument
90 : fwd_pd(GetFCFwdImpl(full_param, is_train, data, weight, bias, out_md)) { in MKLDNNFullyConnectedForward()
107 const NDArray *bias, const mkldnn::memory::desc &out_md);
112 mkldnn::memory::desc *out_md);
|
| H A D | mkldnn_slice.cc | 54 auto out_md = out.GetMKLDNNData()->get_desc(); in MKLDNNSliceFwd() local
59 this->out_ = std::make_shared<mkldnn::memory>(out_md, engine, nullptr); in MKLDNNSliceFwd()
98 auto out_md = out.GetMKLDNNData()->get_desc(); in MKLDNNSlice() local
99 auto out_mem = CreateMKLDNNMem(out, out_md, req); in MKLDNNSlice()
|
| H A D | mkldnn_convolution.cc | 59 auto out_md = GetMemDesc(output); in GetConvFwdImpl() local
146 weight_md, out_md, strides, padding, padding); in GetConvFwdImpl()
150 weight_md, *bias_md_ptr, out_md, strides, padding, in GetConvFwdImpl()
170 weight_md, out_md, strides, dilates, padding, padding); in GetConvFwdImpl()
174 weight_md, *bias_md_ptr, out_md, strides, dilates, in GetConvFwdImpl()
186 auto out_md = GetMemDesc(output); in GetConvBwdData() local
244 weight_md, out_md, strides, padding, padding); in GetConvBwdData()
262 weight_md, out_md, strides, dilates, padding, in GetConvBwdData()
273 auto out_md = GetMemDesc(output); in GetConvBwdWeights() local
331 weight_md, out_md, strides, padding, padding); in GetConvBwdWeights()
[all …]
|
| H A D | mkldnn_deconvolution.cc | 322 out_md(GetMemDesc(out)), in DeconvDescCreator()
344 } else if (out_md.data.format_kind == dnnl_format_kind_any && in ImposePlainWherePadding()
345 out_size != GetMemDescSize(out_md)) { in ImposePlainWherePadding()
346 out_md = GetDesc(out_md, GetDefaultFormat(out_md)); in ImposePlainWherePadding()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | memory.cpp | 258 status_t dnnl_memory_desc_reshape(memory_desc_t *out_md, in dnnl_memory_desc_reshape() argument
269 if (any_null(out_md, in_md) || !memory_desc_sanity_check(in_md) in dnnl_memory_desc_reshape()
282 out_md, ndims, dims, in_md->data_type, format_tag::any); in dnnl_memory_desc_reshape()
452 *out_md = md; in dnnl_memory_desc_reshape()
457 memory_desc_t *out_md, const memory_desc_t *in_md, const int *perm) { in dnnl_memory_desc_permute_axes() argument
458 if (any_null(out_md, in_md) || !memory_desc_sanity_check(in_md) in dnnl_memory_desc_permute_axes()
473 *out_md = *in_md; in dnnl_memory_desc_permute_axes()
476 out_md->dims[perm[d]] = in_md->dims[d]; in dnnl_memory_desc_permute_axes()
477 out_md->padded_dims[perm[d]] = in_md->padded_dims[d]; in dnnl_memory_desc_permute_axes()
478 out_md->padded_offsets[perm[d]] = in_md->padded_offsets[d]; in dnnl_memory_desc_permute_axes()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | memory.cpp | 258 status_t dnnl_memory_desc_reshape(memory_desc_t *out_md, in dnnl_memory_desc_reshape() argument
269 if (any_null(out_md, in_md) || !memory_desc_sanity_check(in_md) in dnnl_memory_desc_reshape()
282 out_md, ndims, dims, in_md->data_type, format_tag::any); in dnnl_memory_desc_reshape()
452 *out_md = md; in dnnl_memory_desc_reshape()
457 memory_desc_t *out_md, const memory_desc_t *in_md, const int *perm) { in dnnl_memory_desc_permute_axes() argument
458 if (any_null(out_md, in_md) || !memory_desc_sanity_check(in_md) in dnnl_memory_desc_permute_axes()
473 *out_md = *in_md; in dnnl_memory_desc_permute_axes()
476 out_md->dims[perm[d]] = in_md->dims[d]; in dnnl_memory_desc_permute_axes()
477 out_md->padded_dims[perm[d]] = in_md->padded_dims[d]; in dnnl_memory_desc_permute_axes()
478 out_md->padded_offsets[perm[d]] = in_md->padded_offsets[d]; in dnnl_memory_desc_permute_axes()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/api/ |
| H A D | test_memory_desc_ops.cpp | 120 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()
125 debug::print_md("expect_out_md", out_md); in Test()
127 ASSERT_EQ(get_out_md, out_md); in Test()
225 void Test(const memory::desc &in_md, const memory::desc &out_md, in Test() argument
232 debug::print_md("expect_out_md", out_md); in Test()
234 ASSERT_EQ(get_out_md, out_md); in Test()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/gtests/api/ |
| H A D | test_memory_desc_ops.cpp | 120 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()
125 debug::print_md("expect_out_md", out_md); in Test()
127 ASSERT_EQ(get_out_md, out_md); in Test()
225 void Test(const memory::desc &in_md, const memory::desc &out_md, in Test() argument
232 debug::print_md("expect_out_md", out_md); in Test()
234 ASSERT_EQ(get_out_md, out_md); in Test()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | convolution_inner_product.cpp | 56 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t *in_md) { in init_conf() argument
60 &out_md, max_dims, dims, in_md->data_type, format_tag::any); in init_conf()
62 out_md = *in_md; in init_conf()
63 out_md.ndims = max_dims; in init_conf()
65 utils::array_copy(&out_md.dims[0], &dims[0], max_dims); in init_conf()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | convolution_inner_product.cpp | 56 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t *in_md) { in init_conf() argument
60 &out_md, max_dims, dims, in_md->data_type, format_tag::any); in init_conf()
62 out_md = *in_md; in init_conf()
63 out_md.ndims = max_dims; in init_conf()
65 utils::array_copy(&out_md.dims[0], &dims[0], max_dims); in init_conf()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ |
| H A D | gpu_inner_product_pd.hpp | 81 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t &in_md) { in template_set_default_params() argument
90 out_md, md_desc_wrapper.blocking_desc()); in template_set_default_params()
92 return memory_desc_init_by_tag(out_md, md_tag); in template_set_default_params()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ |
| H A D | gpu_inner_product_pd.hpp | 81 auto init_md = [&](memory_desc_t &out_md, const memory_desc_t &in_md) { in template_set_default_params() argument
90 out_md, md_desc_wrapper.blocking_desc()); in template_set_default_params()
92 return memory_desc_init_by_tag(out_md, md_tag); in template_set_default_params()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/ |
| H A D | mkldnn_transformer.cc | 235 memory::desc out_md; in Initialize() local
248 out_md = memory::desc(out_dims, memory::data_type::s8, memory::format_tag::abc); in Initialize()
251 out_md = dnnl::memory::desc(out_dims, memory::data_type::f32, memory::format_tag::abc); in Initialize()
257 out_md = dnnl::memory::desc(out_dims, memory::data_type::s32, memory::format_tag::abc); in Initialize()
260 out_md = dnnl::memory::desc(out_dims, memory::data_type::f32, memory::format_tag::abc); in Initialize()
266 auto matmul_d = matmul::desc(query_md, key_md, out_md); in Initialize()
278 cached_out_mem_ = std::make_shared<memory>(out_md, engine, outputs[0].data().dptr<DType>()); in Initialize()
524 dnnl::memory::desc out_md; in Initialize() local
553 out_md = dnnl::memory::desc(dst_dims, get_mkldnn_type(out_dtype), dnnl::memory::format_tag::bac); in Initialize()
556 auto matmul_d = dnnl::matmul::desc(att_md, qkv_md, out_md); in Initialize()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/ |
| H A D | mkldnn_transformer.cc | 235 memory::desc out_md; in Initialize() local
248 out_md = memory::desc(out_dims, memory::data_type::s8, memory::format_tag::abc); in Initialize()
251 out_md = dnnl::memory::desc(out_dims, memory::data_type::f32, memory::format_tag::abc); in Initialize()
257 out_md = dnnl::memory::desc(out_dims, memory::data_type::s32, memory::format_tag::abc); in Initialize()
260 out_md = dnnl::memory::desc(out_dims, memory::data_type::f32, memory::format_tag::abc); in Initialize()
266 auto matmul_d = matmul::desc(query_md, key_md, out_md); in Initialize()
278 cached_out_mem_ = std::make_shared<memory>(out_md, engine, outputs[0].data().dptr<DType>()); in Initialize()
524 dnnl::memory::desc out_md; in Initialize() local
553 out_md = dnnl::memory::desc(dst_dims, get_mkldnn_type(out_dtype), dnnl::memory::format_tag::bac); in Initialize()
556 auto matmul_d = dnnl::matmul::desc(att_md, qkv_md, out_md); in Initialize()
[all …]
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/quantization/mkldnn/ |
| H A D | mkldnn_quantized_fully_connected.cc | 91 mkldnn::memory::desc out_md = GetMemDesc(out_data[fullc::kOut]); in MKLDNNQuantizedFullyConnectedForward() local
92 MKLDNNFCFlattenData(param, out_data[fullc::kOut], &data, &out_md); in MKLDNNQuantizedFullyConnectedForward()
94 param.no_bias ? nullptr : &quantized_bias, out_md); in MKLDNNQuantizedFullyConnectedForward()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/quantization/mkldnn/ |
| H A D | mkldnn_quantized_fully_connected.cc | 91 mkldnn::memory::desc out_md = GetMemDesc(out_data[fullc::kOut]); in MKLDNNQuantizedFullyConnectedForward() local
92 MKLDNNFCFlattenData(param, out_data[fullc::kOut], &data, &out_md); in MKLDNNQuantizedFullyConnectedForward()
94 param.no_bias ? nullptr : &quantized_bias, out_md); in MKLDNNQuantizedFullyConnectedForward()
|