| /dports/science/py-ase/ase-3.22.0/ase/test/md/ |
| H A D | test_verlet_thermostats_asap.py | 44 a_md, traj = prepare_md(a, calculator)
53 with MDalgo(a_md, **kw) as md:
68 thermalize(T_low, a_md, rng)
69 assert abs(a_md.get_temperature() - T_low) < 0.0001
76 temp.append(a_md.get_temperature())
88 a_md = atoms.copy()
89 a_md.calc = calculator
90 traj = Trajectory('Au7Ag.traj', 'w', a_md)
91 return a_md, traj
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | gemm_inner_product.hpp | 69 memory_desc_t a_md, b_md, c_md; in init() local
70 init_2d_desc(&a_md, src_md()); in init()
74 == create_gemm_pd(gemm_pd_, engine, &a_md, &b_md, &c_md, in init()
149 memory_desc_t a_md, b_md, c_md; in init() local
150 init_2d_desc(&a_md, diff_dst_md()); in init()
155 == create_gemm_pd(gemm_pd_, engine, &a_md, &b_md, &c_md, in init()
229 memory_desc_t a_md, b_md, c_md; in init() local
231 init_2d_desc(&a_md, src_md(), true); in init()
235 init_2d_desc(&a_md, diff_dst_md(), true); in init()
242 == create_gemm_pd(gemm_pd_, engine, &a_md, &b_md, &c_md, in init()
|
| H A D | gemm_post_ops_inner_product.hpp | 91 memory_desc_t a_md, b_md, c_md; in init() local
92 init_2d_desc(&a_md, src_md()); in init()
97 == create_gemm_pd(gemm_pd_, engine, &a_md, &b_md, &c_md, in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | gemm_inner_product.hpp | 74 memory_desc_t a_md, b_md, c_md; in init() local
75 init_2d_desc(&a_md, src_md()); in init()
79 == create_gemm_pd(gemm_pd_, engine, &a_md, &b_md, &c_md, in init()
152 memory_desc_t a_md, b_md, c_md; in init() local
153 init_2d_desc(&a_md, diff_dst_md()); in init()
158 == create_gemm_pd(gemm_pd_, engine, &a_md, &b_md, &c_md, in init()
231 memory_desc_t a_md, b_md, c_md; in init() local
233 init_2d_desc(&a_md, src_md(), true); in init()
237 init_2d_desc(&a_md, diff_dst_md(), true); in init()
244 == create_gemm_pd(gemm_pd_, engine, &a_md, &b_md, &c_md, in init()
|
| H A D | gemm_post_ops_inner_product.hpp | 85 memory_desc_t a_md, b_md, c_md; in init() local
86 init_2d_desc(&a_md, src_md()); in init()
91 == create_gemm_pd(gemm_pd_, engine, &a_md, &b_md, &c_md, in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/tutorials/matmul/ |
| H A D | cpu_sgemm_and_matmul.cpp | 133 memory::desc a_md(a_shape, memory::data_type::f32, a_strides); in dynamic_matmul_create() local
147 matmul::desc matmul_d(a_md, b_md, c_md); in dynamic_matmul_create()
197 memory::desc a_md({M, K}, memory::data_type::f32, a_strides); in static_matmul_create_and_execute() local
211 matmul::desc matmul_d(a_md, b_md, c_md); in static_matmul_create_and_execute()
216 memory A_m(a_md, eng, (void *)A); in static_matmul_create_and_execute()
|
| H A D | inference_int8_matmul.cpp | 109 memory::desc a_md({M, K}, memory::data_type::u8, {K, 1}); // M x K layout in matmul_pd_create() local
124 matmul::desc matmul_d(a_md, b_md, c_md); in matmul_pd_create()
|
| H A D | cpu_matmul_quantization.cpp | 271 memory::desc a_md({M, K}, memory::data_type::f32, {K, 1}); in f32_matmul_compute() local
276 memory A_f32_m(a_md, eng, (void *)A_f32.data()); in f32_matmul_compute()
281 matmul::desc matmul_d(a_md, b_md, c_md); in f32_matmul_compute()
|
| /dports/math/onednn/oneDNN-2.5.1/examples/tutorials/matmul/ |
| H A D | cpu_sgemm_and_matmul.cpp | 133 memory::desc a_md(a_shape, memory::data_type::f32, a_strides); in dynamic_matmul_create() local
147 matmul::desc matmul_d(a_md, b_md, c_md); in dynamic_matmul_create()
197 memory::desc a_md({M, K}, memory::data_type::f32, a_strides); in static_matmul_create_and_execute() local
211 matmul::desc matmul_d(a_md, b_md, c_md); in static_matmul_create_and_execute()
216 memory A_m(a_md, eng, (void *)A); in static_matmul_create_and_execute()
|
| H A D | inference_int8_matmul.cpp | 109 memory::desc a_md({M, K}, memory::data_type::u8, {K, 1}); // M x K layout in matmul_pd_create() local
124 matmul::desc matmul_d(a_md, b_md, c_md); in matmul_pd_create()
|
| H A D | cpu_matmul_quantization.cpp | 271 memory::desc a_md({M, K}, memory::data_type::f32, {K, 1}); in f32_matmul_compute() local
276 memory A_f32_m(a_md, eng, (void *)A_f32.data()); in f32_matmul_compute()
281 matmul::desc matmul_d(a_md, b_md, c_md); in f32_matmul_compute()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | gemm_utils.hpp | 106 const memory_desc_t *a_md, const memory_desc_t *b_md, in create_gemm_pd() argument
112 gemm_desc.a_desc = *a_md; in create_gemm_pd()
|
| H A D | rnn.cpp | 54 bool xnor_md(const memory_desc_t *a_md, const memory_desc_t *b_md) { in xnor_md() argument
55 return is_zero_md(a_md) == is_zero_md(b_md); in xnor_md()
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | gemm_utils.hpp | 106 const memory_desc_t *a_md, const memory_desc_t *b_md, in create_gemm_pd() argument
112 gemm_desc.a_desc = *a_md; in create_gemm_pd()
|
| H A D | rnn.cpp | 54 bool xnor_md(const memory_desc_t *a_md, const memory_desc_t *b_md) { in xnor_md() argument
55 return is_zero_md(a_md) == is_zero_md(b_md); in xnor_md()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
| H A D | test_iface_runtime_dims.cpp | 152 memory::desc a_md {{DNNL_RUNTIME_DIM_VAL, 3}, data_type::f32, tag::ab}; in CPU_TEST_F() local
156 CHECK_OK(matmul::desc(a_md, b_md, c_md)); in CPU_TEST_F()
|
| H A D | test_iface_runtime_attr.cpp | 294 memory::desc a_md {{10, 3}, a_dt, tag::ab}; in CPU_TEST_F() local
298 matmul::desc op_d(a_md, b_md, c_md); in CPU_TEST_F()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
| H A D | test_iface_runtime_dims.cpp | 152 memory::desc a_md {{DNNL_RUNTIME_DIM_VAL, 3}, data_type::f32, tag::ab}; in CPU_TEST_F() local
156 CHECK_OK(matmul::desc(a_md, b_md, c_md)); in CPU_TEST_F()
|
| H A D | test_iface_runtime_attr.cpp | 277 memory::desc a_md {{10, 3}, a_dt, tag::ab}; in CPU_TEST_F() local
281 matmul::desc op_d(a_md, b_md, c_md); in CPU_TEST_F()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
| H A D | ref_rnn.cpp | 596 memory_desc_t a_md, b_md, c_md; in init() local
598 create_2d_desc(&a_md, n, k, b_dt, in init()
602 gemm_desc.a_desc = a_md; in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
| H A D | ref_rnn.cpp | 595 memory_desc_t a_md, b_md, c_md; in init() local
597 create_2d_desc(&a_md, n, k, b_dt, in init()
601 gemm_desc.a_desc = a_md; in init()
|