Home
last modified time | relevance | path

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 Dcudnn_softmax_impl.hpp72 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 Dcudnn_softmax_impl.hpp72 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 Didx-vector.cc392 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 Ddim-vector.h489 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 Didx-vector.h286 : 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 Dshiftdim.m67 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 Dgroupby.py185 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 Dswarming.py1182 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 Dliboctave-ChangeLog6531 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.