| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/softmax/ |
| H A D | ref_softmax.cpp | 24 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref_fwd() local
25 get_sizes(prb, outer_size, inner_size, axis_size); in compute_ref_fwd()
35 int64_t ou_in_offset = ou * axis_size * inner_size + in; in compute_ref_fwd()
37 for (int64_t as = 0; as < axis_size; ++as) { in compute_ref_fwd()
42 for (int64_t as = 0; as < axis_size; ++as) { in compute_ref_fwd()
59 for (int64_t as = 0; as < axis_size; ++as) { in compute_ref_fwd()
72 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref_bwd() local
73 get_sizes(prb, outer_size, inner_size, axis_size); in compute_ref_bwd()
83 int64_t ou_in_offset = ou * axis_size * inner_size + in; in compute_ref_bwd()
85 for (int64_t as = 0; as < axis_size; ++as) { in compute_ref_bwd()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/softmax/ |
| H A D | ref_softmax.cpp | 24 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref_fwd() local
25 get_sizes(prb, outer_size, inner_size, axis_size); in compute_ref_fwd()
35 int64_t ou_in_offset = ou * axis_size * inner_size + in; in compute_ref_fwd()
37 for (int64_t as = 0; as < axis_size; ++as) { in compute_ref_fwd()
42 for (int64_t as = 0; as < axis_size; ++as) { in compute_ref_fwd()
59 for (int64_t as = 0; as < axis_size; ++as) { in compute_ref_fwd()
72 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref_bwd() local
73 get_sizes(prb, outer_size, inner_size, axis_size); in compute_ref_bwd()
83 int64_t ou_in_offset = ou * axis_size * inner_size + in; in compute_ref_bwd()
85 for (int64_t as = 0; as < axis_size; ++as) { in compute_ref_bwd()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/concat/ |
| H A D | ref_concat.cpp | 24 int64_t &axis_size) { in get_sizes() argument
30 axis_size = prb->axis_size(); in get_sizes()
35 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref() local
36 get_sizes(prb, outer_size, inner_size, axis_size); in compute_ref()
42 int64_t off_dst = ou * axis_size * inner_size; in compute_ref()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/concat/ |
| H A D | ref_concat.cpp | 24 int64_t &axis_size) { in get_sizes() argument
30 axis_size = prb->axis_size(); in get_sizes()
35 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref() local
36 get_sizes(prb, outer_size, inner_size, axis_size); in compute_ref()
42 int64_t off_dst = ou * axis_size * inner_size; in compute_ref()
|
| /dports/www/chromium-legacy/chromium-88.0.4324.182/third_party/quic_trace/src/tools/render/ |
| H A D | axis_renderer.cc | 79 size_t axis_size, in Ticks() argument
81 size_t max_ticks = axis_size / (max_text_size + 16); in Ticks()
112 {(current - offset) / viewport_size * axis_size, ToString(current)}); in Ticks()
138 const vec2 axis_size = state_->window() - 2 * axis_offset; in Render() local
143 {0, 0, axis_size.x, 0}, in Render()
144 {0, 0, 0, axis_size.y}, in Render()
148 axis_size.x, reference_label_width_)) { in Render()
158 axis_size.y, reference_label_height_)) { in Render()
|
| /dports/www/qt5-webengine/qtwebengine-everywhere-src-5.15.2/src/3rdparty/chromium/third_party/quic_trace/src/tools/render/ |
| H A D | axis_renderer.cc | 79 size_t axis_size, in Ticks() argument
81 size_t max_ticks = axis_size / (max_text_size + 16); in Ticks()
112 {(current - offset) / viewport_size * axis_size, ToString(current)}); in Ticks()
138 const vec2 axis_size = state_->window() - 2 * axis_offset; in Render() local
143 {0, 0, axis_size.x, 0}, in Render()
144 {0, 0, 0, axis_size.y}, in Render()
148 axis_size.x, reference_label_width_)) { in Render()
158 axis_size.y, reference_label_height_)) { in Render()
|
| /dports/math/xgraph/xgraph/ |
| H A D | idraw.c | 23 int axis_size; member
82 axis_family, axis_size, flags, out_info, errmsg) in idrawInit() argument
88 double axis_size; /* Axis font height (points) */
169 out_info -> axis_pad = 2.0 * axis_size;
175 out_info -> tick_len = axis_size;
176 out_info -> axis_height = axis_size;
178 out_info -> axis_width = (axis_size*5.0) / 12.0;
190 idraw_info -> axis_size = axis_size;
290 size = style == T_AXIS ? idraw -> axis_size:
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/shuffle/ |
| H A D | ref_shuffle.cpp | 25 const int64_t axis_size = prb->dims[axis]; in compute_shuffle() local
32 C = axis_size / group_size; in compute_shuffle()
34 R = axis_size / group_size; in compute_shuffle()
46 const size_t dim = axis_size * inner_size; in compute_shuffle()
48 dnnl::impl::parallel_nd(outer_size, axis_size, inner_size, in compute_shuffle()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/shuffle/ |
| H A D | ref_shuffle.cpp | 25 const int64_t axis_size = prb->dims[axis]; in compute_ref() local
32 C = axis_size / group_size; in compute_ref()
34 R = axis_size / group_size; in compute_ref()
46 const size_t dim = axis_size * inner_size; in compute_ref()
48 dnnl::impl::parallel_nd(outer_size, axis_size, inner_size, in compute_ref()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/testing/ |
| H A D | lrn_python.py | 53 axis_size = a_np.shape[axis]
57 sum_end = j+radius+1 if j+radius+1 < axis_size else axis_size
63 sum_end = l+radius+1 if l+radius+1 < axis_size else axis_size
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/testing/ |
| H A D | lrn_python.py | 54 axis_size = a_np.shape[axis]
58 sum_end = j + radius + 1 if j + radius + 1 < axis_size else axis_size
65 sum_end = l + radius + 1 if l + radius + 1 < axis_size else axis_size
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/testing/ |
| H A D | lrn_python.py | 53 axis_size = a_np.shape[axis]
57 sum_end = j+radius+1 if j+radius+1 < axis_size else axis_size
63 sum_end = l+radius+1 if l+radius+1 < axis_size else axis_size
|
| /dports/astro/oskar/OSKAR-2.8.0/oskar/mem/src/ |
| H A D | oskar_mem_read_fits.c | 20 int* num_axes, int** axis_size, double** axis_inc, int* status) in oskar_mem_read_fits() argument
52 if (axis_size) in oskar_mem_read_fits()
54 *axis_size = (int*) realloc(*axis_size, naxis * sizeof(int)); in oskar_mem_read_fits()
65 if (axis_size) (*axis_size)[i] = naxes[i]; in oskar_mem_read_fits()
|
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | batching.py | 31 def batch(fun: lu.WrappedFun, axis_name, axis_size, in_dims, out_dim_dests, argument
35 return _match_axes(batchfun(fun, axis_name, axis_size, in_dims),
36 axis_size, out_dims_thunk, out_dim_dests)
39 def batchfun(axis_name, axis_size, in_dims, *in_vals): argument
46 with core.extend_axis_env(axis_name, axis_size, main):
63 def _match_axes(axis_size, out_dims_thunk, out_dim_dests, *in_vals): argument
253 axis_size, = {x.shape[d] for x, d in zip(in_vals, in_dims)
257 bwd = batch_custom_vjp_bwd(bwd, self.axis_name, axis_size,
286 yield map(partial(matchaxis, axis_size, sum_match=True),
421 avals_in = [core.unmapped_aval(axis_size, 0, aval) if b else aval
[all …]
|
| H A D | pxla.py | 260 axis_size = shape[dim]
267 axis_indices.append(range(axis_size))
268 shard_indices_shape.append(axis_size)
654 axis_size: int,
671 global_axis_size != axis_size):
680 global_axis_size = axis_size
802 if num_local_shards == axis_size:
1229 global_axis_size = axis_size
1582 mesh_mapping = [Replicated(axis_size) for axis_size in axis_sizes.values()]
1742 axis_size = chunk_size * num_devices
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | ref_shuffle.hpp | 69 const int axis_size = pd()->axis_size(); in init() local
72 = pd()->is_fwd() ? group_size : axis_size / group_size; in init()
74 = pd()->is_fwd() ? axis_size / group_size : group_size; in init()
76 axis_size * sizeof(int), platform::get_cache_line_size()); in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | ref_shuffle.hpp | 69 const int axis_size = pd()->axis_size(); in init() local
72 = pd()->is_fwd() ? group_size : axis_size / group_size; in init()
74 = pd()->is_fwd() ? axis_size / group_size : group_size; in init()
76 axis_size * sizeof(int), platform::get_cache_line_size()); in init()
|
| /dports/devel/tinygo/tinygo-0.14.1/llvm-project/mlir/lib/Dialect/QuantOps/Utils/ |
| H A D | FakeQuantSupport.cpp | 134 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
135 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
137 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
153 scales.reserve(axis_size); in fakeQuantAttrsToType()
154 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
155 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|
| /dports/devel/llvm-cheri/llvm-project-37c49ff00e3eadce5d8703fdc4497f28458c64a8/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
144 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
146 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
162 scales.reserve(axis_size); in fakeQuantAttrsToType()
163 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
164 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|
| /dports/devel/llvm-devel/llvm-project-f05c95f10fc1d8171071735af8ad3a9e87633120/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
144 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
146 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
162 scales.reserve(axis_size); in fakeQuantAttrsToType()
163 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
164 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|
| /dports/devel/wasi-compiler-rt12/llvm-project-12.0.1.src/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
144 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
146 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
162 scales.reserve(axis_size); in fakeQuantAttrsToType()
163 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
164 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|
| /dports/devel/wasi-compiler-rt13/llvm-project-13.0.1.src/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
144 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
146 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
162 scales.reserve(axis_size); in fakeQuantAttrsToType()
163 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
164 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|
| /dports/www/chromium-legacy/chromium-88.0.4324.182/third_party/llvm/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
144 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
146 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
162 scales.reserve(axis_size); in fakeQuantAttrsToType()
163 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
164 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|
| /dports/devel/wasi-libcxx/llvm-project-13.0.1.src/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
144 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
146 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
162 scales.reserve(axis_size); in fakeQuantAttrsToType()
163 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
164 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|
| /dports/devel/llvm12/llvm-project-12.0.1.src/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 143 size_t axis_size = rmins.size(); in fakeQuantAttrsToType() local
144 if (axis_size != rmaxs.size()) { in fakeQuantAttrsToType()
146 << axis_size << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()
162 scales.reserve(axis_size); in fakeQuantAttrsToType()
163 zeroPoints.reserve(axis_size); in fakeQuantAttrsToType()
164 for (size_t axis = 0; axis != axis_size; ++axis) { in fakeQuantAttrsToType()
|