Searched defs:array_offset_calculator (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | rnn.hpp | 103 struct array_offset_calculator { struct
107 array_offset_calculator(Telem *base_ptr, Targs... dims) in array_offset_calculator() function
112 array_offset_calculator(const array_offset_calculator<Uelem> &rhs) in array_offset_calculator() function
120 Telem &operator()(Targs... Fargs) const { in operator ()()
124 int64_t nelems() const { in nelems()
131 void set_base_ptr(Telem *base_ptr) { base_ptr_ = base_ptr; } in set_base_ptr()
135 int64_t offset(int64_t d, int64_t pos) const { in offset()
140 int64_t offset(int64_t d, int64_t off, int64_t pos) const { in offset()
145 int64_t offset(int64_t d, int64_t off, int64_t pos, Targs... rem) const { in offset()
149 Telem *base_ptr_;
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | rnn.hpp | 103 struct array_offset_calculator { struct
107 array_offset_calculator(Telem *base_ptr, Targs... dims) in array_offset_calculator() function
112 array_offset_calculator(const array_offset_calculator<Uelem> &rhs) in array_offset_calculator() argument
120 Telem &operator()(Targs... Fargs) const { in operator ()()
124 int64_t nelems() const { in nelems()
131 void set_base_ptr(Telem *base_ptr) { base_ptr_ = base_ptr; } in set_base_ptr()
135 int64_t offset(int64_t d, int64_t pos) const { in offset()
140 int64_t offset(int64_t d, int64_t off, int64_t pos) const { in offset()
145 int64_t offset(int64_t d, int64_t off, int64_t pos, Targs... rem) const { in offset()
149 Telem *base_ptr_;
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | utils.hpp | 414 struct array_offset_calculator { struct
416 array_offset_calculator(Telem *base, Targs... Fargs) : _dims {Fargs...} { in array_offset_calculator() function
424 inline Telem &operator()(Targs... Fargs) const { in operator ()()
431 inline size_t _offset(size_t const dimension, size_t element) const { in _offset()
436 inline size_t _offset( in _offset()
442 inline size_t _offset(size_t const dimension, size_t theta, size_t element, in _offset()
448 Telem *_base_ptr;
449 const int _dims[Tdims];
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | utils.hpp | 434 struct array_offset_calculator { struct
436 array_offset_calculator(Telem *base, Targs... Fargs) : _dims {Fargs...} { in array_offset_calculator() function
444 inline Telem &operator()(Targs... Fargs) const { in operator ()()
451 inline size_t _offset(size_t const dimension, size_t element) const { in _offset()
456 inline size_t _offset( in _offset()
462 inline size_t _offset(size_t const dimension, size_t theta, size_t element, in _offset()
468 Telem *_base_ptr;
469 const int _dims[Tdims];
|