Searched refs:ndim_a (Results 1 – 3 of 3) sorted by relevance
| /dports/devel/py-tinyarray/tinyarray-1.2.4/src/ |
| H A D | arithmetic.cc | 30 int ndim_a, ndim_b; in array_scalar_product() local
32 a->ndim_shape(&ndim_a, &shape_a); in array_scalar_product()
34 assert(ndim_a == 1); in array_scalar_product()
66 int ndim_a, ndim_b; in array_matrix_product() local
68 a->ndim_shape(&ndim_a, &shape_a); in array_matrix_product()
70 assert(ndim_a > 0); in array_matrix_product()
72 int ndim = ndim_a + ndim_b - 2; in array_matrix_product()
140 int ndim_a, ndim_b; in apply_binary_ufunc() local
435 int ndim_a, ndim_b; in dot_product() local
438 if (ndim_a == 0 || ndim_b == 0) { in dot_product()
[all …]
|
| H A D | array.cc | 1088 int ndim_a, ndim_b; in hash() local
1090 reinterpret_cast<Array_base*>(a)->ndim_shape(&ndim_a, &shape_a); in hash()
1094 if (ndim_a != ndim_b) { in hash()
1104 for (int d = 0; d < ndim_a; ++d) { in hash()
1118 equal = compare_data_dtable[int(dtype)](op, a, b, calc_size(ndim_a, shape_a)); in hash()
|
| /dports/science/py-chainer/chainer-7.8.0/onnx_chainer/functions/ |
| H A D | math.py | 210 ndim_a = len(func.inputs[0].shape)
214 if ndim_a > 1 and func.transa:
215 perm = list(range(ndim_a))
|