/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/concat/ |
H A D | ref_concat.cpp | 24 int64_t &axis_size) { in get_sizes() 35 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref() local
|
/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() 35 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref() local
|
/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 39 def batchfun(axis_name, axis_size, in_dims, *in_vals): argument 63 def _match_axes(axis_size, out_dims_thunk, out_dim_dests, *in_vals): argument 277 def batch_custom_vjp_bwd(bwd, axis_name, axis_size, in_dims, out_dim_dests): argument 283 def _match_axes_and_sum(axis_size, out_dims_thunk, out_dim_dests, *in_vals): argument 417 def batch_jaxpr(closed_jaxpr, axis_size, in_batched, instantiate, axis_name): argument 427 def batch_subtrace_instantiate(instantiate, axis_size, main, in_dims, *in_vals): argument 477 def batch_jaxpr(jaxpr, axis_size, in_batched, instantiate, axis_name): argument
|
H A D | pxla.py | 639 def xla_pmap_impl(fun: lu.WrappedFun, *args, backend, axis_name, axis_size, argument 1110 def replicate(val, axis_size, nrep, devices=None, backend=None, in_axis=0): argument 1151 def _pmap_sharding_spec(nrep, axis_size, npart, parts, sharded_aval, map_axis: Optional[int]): argument 1219 in_nodes, name_stack, axis_name, axis_size, argument 1603 def soft_pmap_impl(fun: lu.WrappedFun, *args, axis_name, axis_size, in_axes, out_axes_thunk): argument 1610 def _soft_pmap_callable(fun, axis_name, axis_size, in_axes, out_axes_thunk, *avals): argument 1665 def _soft_pmap_jaxpr(jaxpr, consts, in_axes, axis_name, axis_size, chunk_size): argument 1801 def _pval_to_result_handler(axis_size, nrep, npart, parts, pval, devices, backend): argument
|
/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 72 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref_bwd() local
|
H A D | softmax.cpp | 99 int64_t outer_size = 0, inner_size = 0, axis_size = 0; in fill_data_fwd() local 237 const int64_t axis_size = prb->dims[prb->axis]; in doit() local
|
/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 72 int64_t outer_size {0}, inner_size {0}, axis_size {0}; in compute_ref_bwd() local
|
H A D | softmax.cpp | 101 int64_t outer_size = 0, inner_size = 0, axis_size = 0; in fill_data_fwd() local 240 const int64_t axis_size = prb->dims[prb->axis]; in doit() local
|
/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() 138 const vec2 axis_size = state_->window() - 2 * axis_offset; in Render() local
|
/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() 138 const vec2 axis_size = state_->window() - 2 * axis_offset; in Render() local
|
/dports/misc/mnn/MNN-1.2.0/source/backend/tensorrt/execution/plugin/ |
H A D | LayerNormPlugin.cpp | 14 int axis_size = layer_norm_param->axis()->size(); in LayerNormPlugin() local
|
/dports/math/py-jax/jax-0.2.9/jax/_src/lax/ |
H A D | parallel.py | 860 def _all_gather_via_psum(x, *, all_gather_dimension, axis_name, axis_index_groups, axis_size): argument 869 def _all_gather_impl(x, *, all_gather_dimension, axis_name, axis_index_groups, axis_size): argument 876 …ation_rule(c, x, *, all_gather_dimension, axis_name, axis_index_groups, axis_size, axis_env, platf… argument 891 def _all_gather_abstract_eval(x, *, all_gather_dimension, axis_name, axis_index_groups, axis_size): argument 897 …ll_gather_transpose_rule(cts, x, *, all_gather_dimension, axis_name, axis_index_groups, axis_size): argument 909 …gather_batcher(vals_in, dims_in, *, all_gather_dimension, axis_name, axis_index_groups, axis_size): argument 923 …collective(frame, vals_in, dims_in, all_gather_dimension, axis_name, axis_index_groups, axis_size): argument
|
/dports/misc/mnn/MNN-1.2.0/source/backend/tensorrt/execution/ |
H A D | TRTLayerNorm.cpp | 31 int axis_size = layer_norm_param->axis()->size(); in onEncode() local
|
/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()
|
/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
|
/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
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
H A D | test_shuffle.cpp | 53 const memory::dim axis_size = dims[axis]; in check_shuffle() local 183 const int axis_size = diff_dst_desc.data.dims[p.axis]; in Backward() local
|
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
H A D | test_shuffle.cpp | 53 const memory::dim axis_size = dims[axis]; in check_shuffle() local 183 const int axis_size = diff_dst_desc.data.dims[p.axis]; in Backward() local
|
/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
|
/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
|
H A D | ref_shuffle.cpp | 48 const int axis_size = pd()->axis_size(); in execute_() local
|
/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
|
H A D | ref_shuffle.cpp | 48 const int axis_size = pd()->axis_size(); in execute_() local
|
/dports/graphics/blender/blender-2.91.0/release/scripts/addons/rigify/utils/ |
H A D | widgets_basic.py | 132 def create_pivot_widget(rig, bone_name, axis_size=1.0, cap_size=1.0, square=True, bone_transform_na… argument
|
/dports/misc/mnn/MNN-1.2.0/source/backend/cpu/ |
H A D | CPULayerNorm.cpp | 44 int axis_size = layer_norm_param->axis()->size(); in CPULayerNorm() local
|