Home
last modified time | relevance | path

Searched refs:ndim (Results 1 – 25 of 4310) sorted by relevance

12345678910>>...173

/dports/science/cp2k/cp2k-2e995eec7fd208c8a72d9544807bd8b8ba8cd1cc/src/arnoldi/
H A Darnoldi_geev.F66 lwork = 2*ndim + ndim**2
67 lrwork = 1 + 5*ndim + 2*ndim**2
102 lwork = 2*ndim + ndim**2
103 lrwork = 1 + 5*ndim + 2*ndim**2
136 lwork = 1 + 6*ndim + 2*ndim**2
172 lwork = 1 + 6*ndim + 2*ndim**2
213 diag(ndim) = matrix(ndim, ndim)
255 diag(ndim) = matrix(ndim, ndim)
302 …CALL shseqr('S', 'I', ndim, 1, ndim, tmp_array, ndim, eval1, eval2, evec_r, ndim, work, lwork, inf…
306 …CALL strevc('R', 'B', selects, ndim, tmp_array, ndim, evec_l, ndim, evec_r, ndim, ndim, ndim, work…
[all …]
/dports/science/cp2k-data/cp2k-7.1.0/src/arnoldi/
H A Darnoldi_geev.F66 lwork = 2*ndim + ndim**2
67 lrwork = 1 + 5*ndim + 2*ndim**2
102 lwork = 2*ndim + ndim**2
103 lrwork = 1 + 5*ndim + 2*ndim**2
136 lwork = 1 + 6*ndim + 2*ndim**2
172 lwork = 1 + 6*ndim + 2*ndim**2
213 diag(ndim) = matrix(ndim, ndim)
255 diag(ndim) = matrix(ndim, ndim)
302 …CALL shseqr('S', 'I', ndim, 1, ndim, tmp_array, ndim, eval1, eval2, evec_r, ndim, work, lwork, inf…
306 …CALL strevc('R', 'B', selects, ndim, tmp_array, ndim, evec_l, ndim, evec_r, ndim, ndim, ndim, work…
[all …]
/dports/science/dakota/dakota-6.13.0-release-public.src-UI/packages/external/FSUDace/
H A Dfsu_quality.C305 moment[i1+i2*ndim+j*ndim*ndim] = 0.0; in d_measure()
327 moment[i1+i2*ndim+closest[0]*ndim*ndim] in d_measure()
345 moment[i1+i2*ndim+j*ndim*ndim] in d_measure()
353 moment[i1+i2*ndim+j*ndim*ndim] = moment[i1+i2*ndim+j*ndim*ndim] in d_measure()
377 moment[i+i*ndim+j*ndim*ndim] = moment[i+i*ndim+j*ndim*ndim] in d_measure()
390 a[i1+i2*ndim] = moment[i1+i2*ndim+j*ndim*ndim]; in d_measure()
744 moment[i1+i2*ndim+j*ndim*ndim] = 0.0; in tau_measure()
766 moment[i1+i2*ndim+closest[0]*ndim*ndim] in tau_measure()
784 moment[i1+i2*ndim+j*ndim*ndim] in tau_measure()
792 moment[i1+i2*ndim+j*ndim*ndim] = moment[i1+i2*ndim+j*ndim*ndim] in tau_measure()
[all …]
/dports/audio/codec2/codec2-1.0.3/misc/
H A Dge_train.c113 float err[ndim]; in quantize_ge()
114 float w[ndim]; in quantize_ge()
158 nearest[i] = find_nearest_weighted(codebook, nb_entries, data+i*ndim, weight+i*ndim, ndim); in update_weighted()
170 codebook[n*ndim+j] += w*data[i*ndim+j]; in update_weighted()
252 compute_weights2(data+i*ndim, data+i*ndim, weight+i*ndim, ndim); in main()
254 compute_weights2(data+i*ndim, data+(i-1)*ndim, weight+i*ndim, ndim); in main()
261 pred[i*ndim+j] = data[i*ndim+j] - COEF[j]*data[(i-1)*ndim+j]; in main()
281 quantize_ge(&data[i*ndim], codebook, nb_entries, xq, ndim); in main()
285 delta[i*ndim+j] = xq[j]-data[i*ndim+j]; in main()
286 err[j] += (delta[i*ndim+j])*(delta[i*ndim+j]); in main()
[all …]
/dports/finance/R-cran-ccgarch/ccgarch/R/
H A Dp_mat.R6 A <- diag(para[(ndim+1):(2*ndim)]^2) # ARCH parameter
7 B <- diag(para[(2*ndim+1):(3*ndim)]^2) # GARCH parameter
11 a <- para[1:ndim]^2
12 A <- matrix(para[(ndim+1):(ndim^2+ndim)]^2, ndim, ndim)
13 B <- matrix(para[(ndim^2+ndim+1):(2*ndim^2+ndim)]^2, ndim, ndim)
14 R <- diag(ndim)
15 R[lower.tri(R)] <- para[(2*ndim^2+ndim+1):npara]; R <- (R+t(R)); diag(R) <- 0.5*diag(R)
17 a <- para[1:ndim]^2
18 A <- matrix(para[(ndim+1):(ndim^2+ndim)]^2, ndim, ndim)
19 B <- matrix(para[(ndim^2+ndim+1):(2*ndim^2+ndim)], ndim, ndim)
[all …]
/dports/science/xtb/xtb-6.4.1/src/
H A Dlidep.f9069 real(wp),intent(in) :: A(ndim,ndim)
80 allocate(U(ndim,ndim))
122 real(wp),intent(inout) :: A(ndim,ndim)
167 real(wp),intent(in) :: S(ndim,ndim)
168 real(wp),intent(out) :: X(ndim,ndim)
179 allocate(U(ndim,ndim),ssq(ndim),sisq(ndim))
249 real(wp),intent(in) :: S(ndim,ndim)
250 real(wp),intent(out) :: U(ndim,ndim)
479 real(wp),intent(inout):: H(ndim,ndim)
480 real(wp),intent(in) :: S(ndim,ndim)
[all …]
/dports/finance/R-cran-ccgarch/ccgarch/src/
H A DR_dcc_sim.c7 int i, j, k, nobs = asInteger(n), ndim = LENGTH(a0), cordim = ndim*ndim, info, ione = 1; in dcc_sim() local
18 PROTECT(Q = allocMatrix(REALSXP, ndim, ndim)); in dcc_sim()
19 PROTECT(Qbar = allocMatrix(REALSXP, ndim, ndim)); in dcc_sim()
20 PROTECT(Qlag = allocMatrix(REALSXP, ndim, ndim)); in dcc_sim()
21 PROTECT(diagQ = allocMatrix(REALSXP, ndim, ndim)); in dcc_sim()
22 PROTECT(tmpQ = allocMatrix(REALSXP, ndim, ndim)); in dcc_sim()
23 PROTECT(tmpR = allocMatrix(REALSXP, ndim, ndim)); in dcc_sim()
122 …F77_CALL(dgemm)("N", "T", &ndim, &ndim, &ione, rd_a, rz_row, &ndim, rz_row, &ndim, rd_b, rQlag, &n… in dcc_sim()
130 rdiagQ[(ndim+1)*j] = 1.0/sqrt(rQ[(ndim+1)*j]); in dcc_sim()
132 …F77_CALL(dgemm)("N", "N", &ndim, &ndim, &ndim, &one, rdiagQ, &ndim, rQ, &ndim, &zero, tmprQ, &ndim… in dcc_sim()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/
H A Dbroadcast_reduce-inl.h42 template<int ndim>
43 MSHADOW_XINLINE Shape<ndim> calc_stride(const Shape<ndim>& shape) { in calc_stride()
54 template<int ndim>
69 template<int ndim>
71 Shape<ndim> ret; in unravel()
81 template<int ndim>
91 template<int ndim>
116 template<int ndim>
129 template<int ndim>
369 big.shape_.get<ndim>(), small.shape_.get<ndim>(), in Reduce()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/
H A Dbroadcast_reduce-inl.h42 template<int ndim>
43 MSHADOW_XINLINE Shape<ndim> calc_stride(const Shape<ndim>& shape) { in calc_stride()
54 template<int ndim>
69 template<int ndim>
71 Shape<ndim> ret; in unravel()
81 template<int ndim>
91 template<int ndim>
116 template<int ndim>
129 template<int ndim>
369 big.shape_.get<ndim>(), small.shape_.get<ndim>(), in Reduce()
[all …]
/dports/math/openblas/OpenBLAS-0.3.18/kernel/x86_64/
H A Dcgemm3m_kernel_8x4_haswell.c52 INIT_m8n##ndim\
55 #ndim"881:\n\t"\
58 "prefetcht0 512(%0);" KERNEL_k1m8n##ndim KERNEL_k1m8n##ndim\
59 "prefetcht0 512(%0);" KERNEL_k1m8n##ndim KERNEL_k1m8n##ndim\
62 "prefetcht0 512(%0);" KERNEL_k1m8n##ndim KERNEL_k1m8n##ndim\
63 "prefetcht0 512(%0);" KERNEL_k1m8n##ndim KERNEL_k1m8n##ndim\
67 #ndim"882:\n\t"\
72 #ndim"883:\n\t"\
117 INIT_m4n##ndim\
172 INIT_m2n##ndim\
[all …]
H A Dzgemm3m_kernel_4x4_haswell.c62 #define SAVE_m4(ndim) SAVE_h_m4n##ndim "addq $64,%2;" argument
64 INIT_m4n##ndim\
69 KERNEL_k2m4n##ndim KERNEL_k2m4n##ndim\
71 KERNEL_k2m4n##ndim KERNEL_k2m4n##ndim\
83 SAVE_m4(ndim)
126 #define SAVE_m2(ndim) SAVE_h_m2n##ndim "addq $32,%2;" argument
128 INIT_m2n##ndim\
135 SAVE_m2(ndim)
174 #define SAVE_m1(ndim) SAVE_h_m1n##ndim "addq $16,%2;" argument
176 INIT_m1n##ndim\
[all …]
/dports/math/optpp/optpp-2.4/tests/parallel/
H A Dtstfcn.C45 if (ndim != 2) in init_rosen()
161 if (ndim % 4 != 0) in init_epowell()
228 fi = ndim - sum; in trig()
362 for (i=1; i <= ndim; i++) sum += double(ndim - i + 1)*square(x(i)); in penalty2()
374 for (i = ndim+1; i < 2*ndim; i++) { in penalty2()
447 fn = (3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0; in broyden_tridiag()
711 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0); in broyden1a()
745 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0); in broyden1b()
863 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - xn_plus_1 + 1.0); in tointbroy()
994 int mdim = ndim;
[all …]
/dports/science/dakota/dakota-6.13.0-release-public.src-UI/packages/external/OPTPP/tests/uncon/
H A Dtstfcn.C38 if (ndim != 2) in init_rosen()
89 if (ndim % 4 != 0) in init_epowell()
156 fi = ndim - sum; in trig()
290 for (i=1; i <= ndim; i++) sum += double(ndim - i + 1)*square(x(i)); in penalty2()
302 for (i = ndim+1; i < 2*ndim; i++) { in penalty2()
375 fn = (3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0; in broyden_tridiag()
639 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0); in broyden1a()
673 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0); in broyden1b()
791 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - xn_plus_1 + 1.0); in tointbroy()
922 int mdim = ndim;
[all …]
/dports/science/dakota/dakota-6.13.0-release-public.src-UI/packages/external/OPTPP/tests/parallel/
H A Dtstfcn.C43 if (ndim != 2) in init_rosen()
226 fi = ndim - sum; in trig()
360 for (i=0; i<ndim; i++) sum += double(ndim - i)*square(x(i)); in penalty2()
372 for (i=ndim; i<2*ndim-1; i++) { in penalty2()
445 fn = (3.0 - 2.0*x(ndim-1))*x(ndim-1) - x(ndim-2) - 2.0*xn_plus_1 + 1.0; in broyden_tridiag()
709 fn = fabs((3.0 - 2.0*x(ndim-1))*x(ndim-1) - x(ndim-2) - 2.0*xn_plus_1 + 1.0); in broyden1a()
743 fn = fabs((3.0 - 2.0*x(ndim-1))*x(ndim-1) - x(ndim-2) - 2.0*xn_plus_1 + 1.0); in broyden1b()
861 fn = fabs((3.0 - 2.0*x(ndim-1))*x(ndim-1) - x(ndim-2) - xn_plus_1 + 1.0); in tointbroy()
976 for (int i=0; i<ndim; i++) x(i) = double(i+1)/double(ndim+1); in init_chebyquad()
1028 for (int i=0; i<ndim; i++) x(i) = double(i+1)/double(ndim+1); in init_nelder()
[all …]
/dports/science/qwalk/mainline-1.0.1-300-g1b7e381/src/properties/
H A DAverage_quadrupole.cpp9 Array2 <doublevar> quad(ndim,ndim,0.0); in evaluate()
46 avg.vals.Resize(ndim*ndim*3); in evaluate()
50 avg.vals(ndim*ndim+i*ndim+j)=cos(2*pi*gvecs(i,j)*quad(i,j)); in evaluate()
51 avg.vals(2*ndim*ndim+i*ndim+j)=sin(2*pi*gvecs(i,j)*quad(i,j)); in evaluate()
62 Array2<doublevar> gveclat(ndim,ndim); in read()
69 gvecs.Resize(ndim,ndim); in read()
123 os << avg.vals(k*ndim*ndim+i*ndim+j) << " "; in write_summary()
130 os << err.vals(k*ndim*ndim+i*ndim+j) << " "; in write_summary()
153 Array2 <doublevar> tmp(ndim,ndim); in jsonOutput()
159 tmp(i,j)=avg.vals(k*ndim*ndim+i*ndim+j); in jsonOutput()
[all …]
/dports/science/afni/afni-AFNI_21.3.16/src/
H A Drcov.c4 int ndim ; member
15 register int ndim=cv->ndim , ii,jj ; in forward_solve_inplace() local
30 register int ndim=cv->ndim , ii,jj ; in backward_solve_inplace() local
45 register int ndim=cv->ndim , ii,jj,kk ; in compute_choleski() local
51 cv->cfac = (float *) malloc(sizeof(float)*ndim*ndim) ; in compute_choleski()
84 if( ndim < 2 || nvec < ndim || vec == NULL ) return NULL ; in robust_covar()
87 cv->ndim = ndim ; in robust_covar()
111 for( ii=0 ; ii < ndim ; ii++ ) nmat[ii+jj*ndim] = 0.0 ; in robust_covar()
122 nmat[jj+ii*ndim] = (nmat[ii+jj*ndim] *= fnvec) ; in robust_covar()
143 for( ii=0 ; ii < ndim ; ii++ ) nmat[ii+jj*ndim] = 0.0 ; in robust_covar()
[all …]
/dports/math/optpp/optpp-2.4/tests/constraints/
H A Dtstfcn.C44 if (ndim != 2) in init_rosen()
153 if (ndim % 4 != 0) in init_epowell()
220 fi = ndim - sum; in trig()
244 fi = ndim - sum; in trig_as_a_constraint()
375 for (i=1; i <= ndim; i++) sum += double(ndim - i + 1)*square(x(i)); in penalty2()
387 for (i = ndim+1; i < 2*ndim; i++) { in penalty2()
460 fn = (3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0; in broyden_tridiag()
724 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0); in broyden1a()
758 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0); in broyden1b()
876 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - xn_plus_1 + 1.0); in tointbroy()
[all …]
/dports/science/dakota/dakota-6.13.0-release-public.src-UI/packages/external/OPTPP/tests/constraints/
H A Dtstfcn.C42 if (ndim != 2) in init_rosen()
151 if (ndim % 4 != 0) in init_epowell()
218 fi = ndim - sum; in trig()
242 fi = ndim - sum; in trig_as_a_constraint()
387 for (i = ndim+1; i < 2*ndim; i++) { in penalty2()
460 fn = (3.0 - 2.0*x(ndim-1))*x(ndim-1) - x(ndim-1-1) - 2.0*xn_plus_1 + 1.0; in broyden_tridiag()
724 fn = fabs((3.0 - 2.0*x(ndim-1))*x(ndim-1) - x(ndim-1-1) - 2.0*xn_plus_1 + 1.0); in broyden1a()
758 fn = fabs((3.0 - 2.0*x(ndim-1))*x(ndim-1) - x(ndim-1-1) - 2.0*xn_plus_1 + 1.0); in broyden1b()
876 fn = fabs((3.0 - 2.0*x(ndim-1))*x(ndim-1) - x(ndim-1-1) - xn_plus_1 + 1.0); in tointbroy()
1007 int mdim = ndim;
[all …]
/dports/science/nwchem-data/nwchem-7.0.2-release/contrib/openmp_tests/dgemm/
H A Ddgemm_ga.F94 ndim = -1
125 stack = 3*ndim*ndim+3*(ndim*ndim+ncore-1)/ncore
128 stack = 4*(ndim*ndim+ncore-1)/ncore
135 if (ndim.le.0) call ga_error("illegal dimension",ndim)
183 call random_number(dbl_mb(k_A1:k_A1-1+ndim*ndim))
184 call random_number(dbl_mb(k_B1:k_B1-1+ndim*ndim))
207 call ga_put(g_A,1,ndim,1,ndim,dbl_mb(k_A1),ndim)
208 call ga_put(g_B,1,ndim,1,ndim,dbl_mb(k_B1),ndim)
245 call ga_get(g_C,1,ndim,1,ndim,dbl_mb(k_C1),ndim)
250 call dgemm_i8('N','N',ndim,ndim,ndim,-1.0_w,dbl_mb(k_A1),ndim,
[all …]
/dports/science/nwchem/nwchem-7b21660b82ebd85ef659f6fba7e1e73433b0bd0a/contrib/openmp_tests/dgemm/
H A Ddgemm_ga.F94 ndim = -1
125 stack = 3*ndim*ndim+3*(ndim*ndim+ncore-1)/ncore
128 stack = 4*(ndim*ndim+ncore-1)/ncore
135 if (ndim.le.0) call ga_error("illegal dimension",ndim)
183 call random_number(dbl_mb(k_A1:k_A1-1+ndim*ndim))
184 call random_number(dbl_mb(k_B1:k_B1-1+ndim*ndim))
207 call ga_put(g_A,1,ndim,1,ndim,dbl_mb(k_A1),ndim)
208 call ga_put(g_B,1,ndim,1,ndim,dbl_mb(k_B1),ndim)
245 call ga_get(g_C,1,ndim,1,ndim,dbl_mb(k_C1),ndim)
250 call dgemm_i8('N','N',ndim,ndim,ndim,-1.0_w,dbl_mb(k_A1),ndim,
[all …]
/dports/science/dakota/dakota-6.13.0-release-public.src-UI/packages/external/OPTPP/tests/hock/
H A Dtstfcn.C44 if (ndim != 2) in init_rosen()
160 if (ndim % 4 != 0) in init_epowell()
227 fi = ndim - sum; in trig()
378 for (i=1; i <= ndim; i++) sum += double(ndim - i + 1)*square(x(i)); in penalty2()
390 for (i = ndim+1; i < 2*ndim; i++) { in penalty2()
463 fn = (3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0; in broyden_tridiag()
727 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0); in broyden1a()
761 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - 2.0*xn_plus_1 + 1.0); in broyden1b()
879 fn = fabs((3.0 - 2.0*x(ndim))*x(ndim) - x(ndim-1) - xn_plus_1 + 1.0); in tointbroy()
1010 int mdim = ndim;
[all …]
/dports/cad/tochnog/Sources/Tochnog-Latest-jan-2014/
H A Dpolynom.cc414 matrix_ab( &p[ipoint*ndim*nnol], old_coord, xj, ndim, nnol, ndim ); in pol()
415 if ( !matrix_inverse( xj, &xj_inv[ipoint*ndim*ndim], detj, ndim ) ) { in pol()
417 array_set( &xj_inv[ipoint*ndim*ndim], 0., ndim*ndim ); in pol()
419 matrix_ab( &xj_inv[ipoint*ndim*ndim], &p[ipoint*ndim*nnol], in pol()
420 &old_d[ipoint*ndim*nnol], ndim, ndim, nnol ); in pol()
421 matrix_ab( &p[ipoint*ndim*nnol], new_coord, xj, ndim, nnol, ndim ); in pol()
422 if ( !matrix_inverse( xj, &xj_inv[ipoint*ndim*ndim], detj, ndim ) ) { in pol()
424 array_set( &xj_inv[ipoint*ndim*ndim], 0., ndim*ndim ); in pol()
426 matrix_ab( &xj_inv[ipoint*ndim*ndim], &p[ipoint*ndim*nnol], in pol()
427 &new_d[ipoint*ndim*nnol], ndim, ndim, nnol ); in pol()
[all …]
/dports/science/berkeleygw/BGW-2.0.0/Common/qhull/
H A Dlibtile_qhull_find.c264 T = (double*) calloc(ndim*ndim, sizeof(double)); in init_barycentric_transforms()
269 work = (double*) malloc(((ndim<4)?4:ndim)*ndim*sizeof(double)); in init_barycentric_transforms()
288 cur_transf[ndim*ndim + i] = *last_coord; in init_barycentric_transforms()
357 cur_transf += ndim*(ndim+1); in init_barycentric_transforms()
403 if (i==ndim) { in get_barycentric_coord_single()
405 for (j=0; j<ndim; j++) c[ndim] -= c[j]; in get_barycentric_coord_single()
409 c[i] += cur_transf[ndim*i + j] * (x[j] - cur_transf[ndim*ndim + j]); in get_barycentric_coord_single()
429 c[ndim] = 1.0; in is_barycentric_inside()
433 c[i] += cur_transf[ndim*i + j] * (x[j] - cur_transf[ndim*ndim + j]); in is_barycentric_inside()
465 cur_transf += ndim*(ndim+1); in find_simplex_bruteforce()
[all …]
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/
H A Dnp_matmul_op-inl.h42 for (int i = ashape.ndim() - 3, j = bshape.ndim() - 3; in MatmulNeedBroadcast()
91 CHECK_LT(dimend, ndim) << "dimend must be < " << ndim in GetBroadcastKernelShape()
274 size_t ndim = out_shape.ndim(); in NumpyMatmulForward() local
276 ndim = 2; in NumpyMatmulForward()
280 ndim = out_shape.ndim() + 1; in NumpyMatmulForward()
286 newshape[ndim - 1] = out_shape[ndim - 2]; in NumpyMatmulForward()
289 ndim = out_shape.ndim() + 1; in NumpyMatmulForward()
359 size_t ndim = out_shape.ndim(); in NumpyMatmulBackward() local
365 ndim = 2; in NumpyMatmulBackward()
373 ndim = out_shape.ndim() + 1; in NumpyMatmulBackward()
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/
H A Dnp_matmul_op-inl.h42 for (int i = ashape.ndim() - 3, j = bshape.ndim() - 3; in MatmulNeedBroadcast()
91 CHECK_LT(dimend, ndim) << "dimend must be < " << ndim in GetBroadcastKernelShape()
274 size_t ndim = out_shape.ndim(); in NumpyMatmulForward() local
276 ndim = 2; in NumpyMatmulForward()
280 ndim = out_shape.ndim() + 1; in NumpyMatmulForward()
286 newshape[ndim - 1] = out_shape[ndim - 2]; in NumpyMatmulForward()
289 ndim = out_shape.ndim() + 1; in NumpyMatmulForward()
359 size_t ndim = out_shape.ndim(); in NumpyMatmulBackward() local
365 ndim = 2; in NumpyMatmulBackward()
373 ndim = out_shape.ndim() + 1; in NumpyMatmulBackward()
[all …]

12345678910>>...173