| /dports/devel/py-pyopencl/pyopencl-2018.2/pyopencl/ |
| H A D | reduction.py | 508 def _get_dot_expr(dtype_out, dtype_a, dtype_b, conjugate_first, argument
511 if dtype_a is None:
514 dtype_b = dtype_a
519 dtype_a.type(0), dtype_b.type(0),
522 a_is_complex = dtype_a.kind == "c"
530 if a_is_complex and (dtype_a != dtype_out):
552 def get_dot_kernel(ctx, dtype_out, dtype_a=None, dtype_b=None, argument
556 dtype_out, dtype_a, dtype_b, conjugate_first,
572 "tp_a": dtype_to_ctype(dtype_a),
582 dtype_out, dtype_a, dtype_b, conjugate_first,
[all …]
|
| H A D | elementwise.py | 591 def get_axpbz_kernel(context, dtype_a, dtype_x, dtype_b, dtype_z): argument
592 a_is_complex = dtype_a.kind == "c"
607 if dtype_a != dtype_z:
617 if dtype_a != dtype_z:
642 "tp_a": dtype_to_ctype(dtype_a),
859 VectorArg(dtype_a, "a", with_offset=True),
870 VectorArg(dtype_a, "a", with_offset=True),
878 def get_array_scalar_comparison_kernel(context, operator, dtype_a): argument
881 VectorArg(dtype_a, "a", with_offset=True),
882 ScalarArg(dtype_a, "b"),
[all …]
|
| /dports/math/py-pandas/pandas-1.2.5/pandas/tests/frame/ |
| H A D | test_ufunc.py | 98 def test_binary_input_aligns_columns(dtype_a, dtype_b): argument
100 pd.api.types.is_extension_array_dtype(dtype_a)
101 or isinstance(dtype_a, dict)
107 df1 = pd.DataFrame({"A": [1, 2], "B": [3, 4]}).astype(dtype_a)
109 if isinstance(dtype_a, dict) and isinstance(dtype_b, dict):
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/arm_cpu/ |
| H A D | tensor_intrin.py | 83 dtype_a = '%s8x%d' % (dtype, num_int8_elements)
87 a_int8 = ins[0].vload([0], dtype_a)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/arm_cpu/ |
| H A D | tensor_intrin.py | 83 dtype_a = '%s8x%d' % (dtype, num_int8_elements)
87 a_int8 = ins[0].vload([0], dtype_a)
|
| /dports/science/nwchem/nwchem-7b21660b82ebd85ef659f6fba7e1e73433b0bd0a/src/nwdft/rt_tddft/matutils/ |
| H A D | matutils.F | 586 integer dtype_a, dtype_b
607 call ga_inquire (g_a, dtype_a, n1_a, n2_a)
615 if (dtype_a .eq. dtype_b) then !they are the same data type
617 if (dtype_a .eq. mt_dbl) then ! real data
637 elseif (dtype_a .eq. mt_dcpl) then ! complex data
|
| /dports/science/nwchem-data/nwchem-7.0.2-release/src/nwdft/rt_tddft/matutils/ |
| H A D | matutils.F | 586 integer dtype_a, dtype_b
607 call ga_inquire (g_a, dtype_a, n1_a, n2_a)
615 if (dtype_a .eq. dtype_b) then !they are the same data type
617 if (dtype_a .eq. mt_dbl) then ! real data
637 elseif (dtype_a .eq. mt_dcpl) then ! complex data
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/arm_cpu/ |
| H A D | tensor_intrin.py | 555 dtype_a = "%s8x%d" % (dtype, num_int8_elements)
559 a_int8 = ins[0].vload([0], dtype_a)
|
| /dports/devel/py-tinyarray/tinyarray-1.2.4/src/ |
| H A D | array.cc | 1671 Dtype dtype_a = NONE, dtype_b = NONE, dtype; in coerce_to_arrays() local
1672 a = array_from_arraylike(a, &dtype_a); in coerce_to_arrays()
1674 b = array_from_arraylike(b, &dtype_b, dtype_a); in coerce_to_arrays()
1681 dtype = Dtype(std::max(int(dtype_a), int(dtype_b))); in coerce_to_arrays()
1682 if (dtype_a != dtype) { in coerce_to_arrays()
1683 PyObject *temp = convert_array(dtype, a, dtype_a); in coerce_to_arrays()
|
| /dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/routines/ |
| H A D | creation.cc | 319 Dtype dtype_a = dtype.value_or(internal::GetDefaultDtype(chainerx::DtypeKind::kFloat)); in Linspace() local
326 Array out = Empty(Shape{num_a}, dtype_a, device); in Linspace()
|
| /dports/devel/py-pyopencl/pyopencl-2018.2/test/ |
| H A D | test_array.py | 125 for dtype_a, dtype_b in [
135 ary_a = make_random_array(queue, dtype_a, 1).get()[0]
138 ary_a = make_random_array(queue, dtype_a, size)
148 print(op, dtype_a, dtype_b, is_scalar_a, is_scalar_b)
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
| H A D | test_operator.py | 8542 def assert_same_dtype(dtype_a, dtype_b): argument
8550 assert dtype_a == dtype_b,\
8551 TypeError('Unmatched data types: %s vs %s' % (dtype_a, dtype_b))
8713 def assert_same_dtype(dtype_a, dtype_b): argument
8721 assert dtype_a == dtype_b,\
8722 TypeError('Unmatched data types: %s vs %s' % (dtype_a, dtype_b))
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
| H A D | test_operator.py | 8542 def assert_same_dtype(dtype_a, dtype_b): argument
8550 assert dtype_a == dtype_b,\
8551 TypeError('Unmatched data types: %s vs %s' % (dtype_a, dtype_b))
8713 def assert_same_dtype(dtype_a, dtype_b): argument
8721 assert dtype_a == dtype_b,\
8722 TypeError('Unmatched data types: %s vs %s' % (dtype_a, dtype_b))
|