Searched refs:orig_dims (Results 1 – 9 of 9) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/ |
H A D | cudnn_softmax_impl.hpp | 72 modified_dims[i] = orig_dims[i]; in convert_dims_softmax() 75 modified_dims[3] *= orig_dims[i]; in convert_dims_softmax() 82 modified_dims[0] = orig_dims[1]; in convert_dims_softmax() 83 modified_dims[1] = orig_dims[0]; in convert_dims_softmax() 90 modified_dims[2] = orig_dims[1]; in convert_dims_softmax() 112 modified_dims[1] = orig_dims[0]; in convert_dims_softmax() 118 modified_dims[0] = orig_dims[0]; in convert_dims_softmax() 119 modified_dims[1] = orig_dims[2]; in convert_dims_softmax() 123 modified_dims[3] = orig_dims[1]; in convert_dims_softmax() 126 modified_dims[0] = orig_dims[0] * orig_dims[2]; in convert_dims_softmax() [all …]
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/ |
H A D | cudnn_softmax_impl.hpp | 72 modified_dims[i] = orig_dims[i]; in convert_dims_softmax() 75 modified_dims[3] *= orig_dims[i]; in convert_dims_softmax() 82 modified_dims[0] = orig_dims[1]; in convert_dims_softmax() 83 modified_dims[1] = orig_dims[0]; in convert_dims_softmax() 90 modified_dims[2] = orig_dims[1]; in convert_dims_softmax() 112 modified_dims[1] = orig_dims[0]; in convert_dims_softmax() 118 modified_dims[0] = orig_dims[0]; in convert_dims_softmax() 119 modified_dims[1] = orig_dims[2]; in convert_dims_softmax() 123 modified_dims[3] = orig_dims[1]; in convert_dims_softmax() 126 modified_dims[0] = orig_dims[0] * orig_dims[2]; in convert_dims_softmax() [all …]
|
/dports/math/octave/octave-6.4.0/liboctave/array/ |
H A D | idx-vector.cc | 392 orig_dims () in idx_vector_rep() 399 orig_dims = dv.make_nd_vector (len); in idx_vector_rep() 420 aowner (nullptr), orig_dims () in idx_vector_rep() 424 orig_dims = dv.make_nd_vector (len); in idx_vector_rep() 491 if (new_rep->orig_dims.ndims () == 2 && new_rep->orig_dims(0) == 1) in sort_uniq_clone() 509 if (new_rep->orig_dims.ndims () == 2 && new_rep->orig_dims(0) == 1) in sort_uniq_clone() 551 idx.clear (orig_dims); in sort_idx() 572 idx.clear (orig_dims); in sort_idx() 616 Array<double> retval (orig_dims); in unconvert() 673 orig_dims = dv.make_nd_vector (len); in idx_mask_rep() [all …]
|
H A D | dim-vector.h | 489 dim_vector orig_dims; in make_nd_vector() local 493 orig_dims = *this; in make_nd_vector() 495 for (int i = 0; i < orig_dims.ndims (); i++) in make_nd_vector() 497 if (orig_dims(i) != 1) in make_nd_vector() 499 orig_dims(i) = n; in make_nd_vector() 505 orig_dims = dim_vector (n, 1); in make_nd_vector() 507 return orig_dims; in make_nd_vector()
|
H A D | idx-vector.h | 286 : data (nullptr), len (0), ext (0), aowner (nullptr), orig_dims () in idx_vector_rep() 293 aowner (nullptr), orig_dims (od) in idx_vector_rep() 334 dim_vector orig_dimensions (void) const { return orig_dims; } in orig_dimensions() 359 dim_vector orig_dims; variable 373 lsti (-1), lste (-1), aowner (nullptr), orig_dims (od) in idx_mask_rep() 404 dim_vector orig_dimensions (void) const { return orig_dims; } in orig_dimensions() 437 dim_vector orig_dims; variable
|
/dports/math/octave/octave-6.4.0/scripts/general/ |
H A D | shiftdim.m | 67 orig_dims = size (x); variable 71 (n = find (orig_dims != 1, 1) - 1) || (n = nd); 82 y = reshape (x, [singleton_dims, orig_dims]);
|
/dports/devel/py-xarray/xarray-0.20.1/xarray/core/ |
H A D | groupby.py | 185 orig_dims = group.dims 186 stacked_dim = "stacked_" + "_".join(orig_dims) 191 group = group.stack(**{stacked_dim: orig_dims}).copy() 192 obj = obj.stack(**{stacked_dim: orig_dims})
|
/dports/www/chromium-legacy/chromium-88.0.4324.182/tools/swarming_client/ |
H A D | swarming.py | 1182 orig_dims = [{ 1186 orig_dims.sort(key=lambda x: (x['key'], x['value'])) 1199 dimensions=orig_dims, 1246 dims = list(orig_dims)
|
/dports/math/octave/octave-6.4.0/etc/OLD-ChangeLogs/ |
H A D | liboctave-ChangeLog | 6531 set one_zero to 0 and orig_dims to size of resulting index vector. 10652 * idx-vector.h (idx_vector::orig_empty): Check orig_dims for 10654 (idx_vector::idx_vector_rep::orig_rows): Define using orig_dims. 11148 * idx-vector.h (idx_vector::orig_dims): New member variable.
|