Home
last modified time | relevance | path

Searched refs:split_axis (Results 1 – 25 of 66) sorted by relevance

123

/dports/science/chrono/chrono-7.0.1/src/chrono_distributed/physics/
H A DChDomainDistributed.cpp32 split_axis = 0; in ChDomainDistributed()
42 split_axis = i; in SetSplitAxis()
64 split_axis = (len_x >= len_y) ? 0 : 1; in SetSimDomain()
65 split_axis = (len_z >= boxhi[split_axis] - boxlo[split_axis]) ? 2 : split_axis; in SetSimDomain()
73 double sub_len = (boxhi[split_axis] - boxlo[split_axis]) / my_sys->num_ranks; in SplitDomain()
76 if (split_axis == i) { in SplitDomain()
88 double sub_len = subhi[split_axis] - sublo[split_axis]; in GetRank()
89 double pt = pos[split_axis] - boxlo[split_axis]; in GetRank()
101 double high = subhi[split_axis]; in GetRegion()
102 double low = sublo[split_axis]; in GetRegion()
[all …]
H A DChDomainDistributed.h137 int GetSplitAxis() const { return split_axis; } in GetSplitAxis()
157 int split_axis; ///< Index of the dimension of the longest edge of the global domain variable
/dports/math/fcl05/fcl-0.5.0/include/fcl/BVH/
H A DBV_splitter.h116 return q[split_axis] > split_value; in apply()
131 int split_axis;
160 split_axis = axis; in computeRule_bvcenter()
174 split_axis = axis; in computeRule_mean()
182 … sum += (vertices[t[0]][split_axis] + vertices[t[1]][split_axis] + vertices[t[2]][split_axis]); in computeRule_mean()
191 sum += vertices[primitive_indices[i]][split_axis]; in computeRule_mean()
208 split_axis = axis; in computeRule_median()
216 …proj[i] = (vertices[t[0]][split_axis] + vertices[t[1]][split_axis] + vertices[t[2]][split_axis]) /… in computeRule_median()
222 proj[i] = vertices[primitive_indices[i]][split_axis]; in computeRule_median()
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/loss/
H A Dcrf1d.py5 from chainer.functions.array import split_axis
118 alpha, alpha_rest = split_axis.split_axis(alpha, [batch], axis=0)
135 y_prev, _ = split_axis.split_axis(y_prev, [batch], axis=0)
136 score, score_rest = split_axis.split_axis(score, [batch], axis=0)
186 alpha, alpha_rest = split_axis.split_axis(alpha, [batch], axis=0)
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/primitives/
H A Dmvn.hpp35 std::size_t split_axis; member
52 split_axis = config.split_axis; in MVNOp()
61 for (int j = 0; j < split_axis; j++) in MVNOp()
88 auto outer_size = input.size_range(0, split_axis); in forward()
89 auto inner_size = input.size_range(split_axis, input.rank()); in forward()
127 std::size_t split_axis; member in cv::dnn::cuda4dnn::MVNOp
/dports/math/py-jax/jax-0.2.9/jax/_src/lax/
H A Dparallel.py310 return all_to_all_p.bind(x, split_axis=split_axis, concat_axis=concat_axis,
709 split_axis=split_axis, concat_axis=concat_axis, axis_name=axis_name,
716 if concat_axis == split_axis:
721 split_axis += 1
733 split_axis=concat_axis,
734 concat_axis=split_axis,
741 if d <= split_axis:
742 split_axis += 1
754 split_axis=split_axis,
766 split_axis_adj = split_axis + (1 if d <= split_axis else 0)
[all …]
/dports/science/py-chainer/chainer-7.8.0/chainer/links/rnn/
H A Dtree_lstm.py6 from chainer.functions.array import split_axis
100 W_x_aio_in, W_x_f_in = split_axis.split_axis(
105 a, i, o = split_axis.split_axis(aio_in, 3, axis=1)
116 W_h_fs_in = concat.concat(split_axis.split_axis(
H A Dpeephole.py5 from chainer.functions.array import split_axis
101 a, i, f, o = split_axis.split_axis(lstm_in, 4, 2)
H A Dzoneoutlstm.py5 from chainer.functions.array import split_axis
91 a, i, f, o = split_axis.split_axis(lstm_in, 4, 2)
H A Dlstm.py5 from chainer.functions.array import split_axis
300 h_update, h_rest = split_axis.split_axis(
/dports/math/fcl/fcl-0.7.0/include/fcl/geometry/bvh/detail/
H A DBV_splitter-inl.h104 return q[splitter.split_axis] > splitter.split_value; in run()
133 splitter.split_axis = axis; in run()
164 splitter.split_axis = axis; in run()
172 sum += splitter.vertices[t[0]][splitter.split_axis] in run()
173 + splitter.vertices[t[1]][splitter.split_axis] in run()
174 + splitter.vertices[t[2]][splitter.split_axis]; in run()
183 sum += splitter.vertices[primitive_indices[i]][splitter.split_axis]; in run()
217 splitter.split_axis = axis; in run()
225 proj[i] = (splitter.vertices[t[0]][splitter.split_axis] in run()
226 + splitter.vertices[t[1]][splitter.split_axis] in run()
[all …]
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/rnn/
H A Dn_step_gru.py9 from chainer.functions.array import split_axis
382 ys = chainer.functions.split_axis(ys, sections, 0)
401 W_r_x, W_z_x, W_x = split_axis.split_axis(gru_x, 3, axis=1)
402 U_r_h, U_z_h, U_x = split_axis.split_axis(gru_h, 3, axis=1)
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/math/
H A Dcumprod.py55 _, x = F.split_axis(x, (1,), axis)
57 y, ylast = F.split_axis(y, (-1,), axis)
100 _, gxmul = F.split_axis(gxadd, (1,), axis)
101 y, _ = F.split_axis(y, (-1,), axis)
/dports/graphics/ogre3d19/sinbad-ogre-dd30349ea667/OgreMain/src/
H A DOgreSubMesh.cpp289 Cluster split (int split_axis, const VertexElement *poselem, in split()
292 Real r = (mMin [split_axis] + mMax [split_axis]) * 0.5f; in split()
301 if (v [split_axis] > r) in split()
399 int split_axis = 0; in generateExtremes() local
407 split_axis = i; in generateExtremes()
412 boxes.push_back (split_box->split (split_axis, poselem, vdata, vsz)); in generateExtremes()
/dports/graphics/ogre3d/ogre-1.11.6/OgreMain/src/
H A DOgreSubMesh.cpp282 Cluster split (int split_axis, const VertexElement *poselem, in split()
285 Real r = (mMin [split_axis] + mMax [split_axis]) * 0.5f; in split()
294 if (v [split_axis] > r) in split()
395 int split_axis = 0; in generateExtremes() local
403 split_axis = i; in generateExtremes()
408 boxes.push_back (split_box->split (split_axis, poselem, vdata, vsz)); in generateExtremes()
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/array/
H A Dvstack.py42 gxs = chainer.functions.split_axis(gy, len(self.inputs), 0)
46 return chainer.functions.split_axis(gy, sizes, 0)
H A Dhstack.py45 gx = chainer.functions.split_axis(gy, len(self.inputs), 0)
50 return chainer.functions.split_axis(gy, sizes, axis)
H A Ddstack.py46 gxs = chainer.functions.split_axis(gy, len(self.inputs), axis=2)
50 return chainer.functions.split_axis(gy, sizes, axis=2)
/dports/databases/percona57-pam-for-mysql/boost_1_59_0/boost/geometry/index/detail/rtree/rstar/
H A Dredistribute_elements.hpp404 size_t split_axis = 0; in apply() local
418 split_axis, split_corner, split_index, in apply()
423 BOOST_GEOMETRY_INDEX_ASSERT(split_axis < dimension, "unexpected value"); in apply()
431 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
436 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
/dports/databases/percona57-server/boost_1_59_0/boost/geometry/index/detail/rtree/rstar/
H A Dredistribute_elements.hpp404 size_t split_axis = 0; in apply() local
418 split_axis, split_corner, split_index, in apply()
423 BOOST_GEOMETRY_INDEX_ASSERT(split_axis < dimension, "unexpected value"); in apply()
431 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
436 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
/dports/databases/xtrabackup/boost_1_59_0/boost/geometry/index/detail/rtree/rstar/
H A Dredistribute_elements.hpp404 size_t split_axis = 0; in apply() local
418 split_axis, split_corner, split_index, in apply()
423 BOOST_GEOMETRY_INDEX_ASSERT(split_axis < dimension, "unexpected value"); in apply()
431 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
436 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
/dports/math/deal.ii/dealii-803d21ff957e349b3799cd3ef2c840bc78734305/bundled/boost-1.70.0/include/boost/geometry/index/detail/rtree/rstar/
H A Dredistribute_elements.hpp408 size_t split_axis = 0; in apply() local
422 split_axis, split_corner, split_index, in apply()
427 BOOST_GEOMETRY_INDEX_ASSERT(split_axis < dimension, "unexpected value"); in apply()
435 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
440 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
/dports/databases/percona57-client/boost_1_59_0/boost/geometry/index/detail/rtree/rstar/
H A Dredistribute_elements.hpp404 size_t split_axis = 0; in apply() local
418 split_axis, split_corner, split_index, in apply()
423 BOOST_GEOMETRY_INDEX_ASSERT(split_axis < dimension, "unexpected value"); in apply()
431 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
436 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
/dports/security/keybase/client-v5.7.1/shared/ios/Pods/boost-for-react-native/boost/geometry/index/detail/rtree/rstar/
H A Dredistribute_elements.hpp404 size_t split_axis = 0; in apply() local
418 split_axis, split_corner, split_index, in apply()
423 BOOST_GEOMETRY_INDEX_ASSERT(split_axis < dimension, "unexpected value"); in apply()
431 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
436 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
/dports/databases/mysqlwsrep57-server/boost_1_59_0/boost/geometry/index/detail/rtree/rstar/
H A Dredistribute_elements.hpp404 size_t split_axis = 0; in apply() local
418 split_axis, split_corner, split_index, in apply()
423 BOOST_GEOMETRY_INDEX_ASSERT(split_axis < dimension, "unexpected value"); in apply()
431 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()
436 …::apply(elements_copy, split_axis, split_index, translator); // MAY THR… in apply()

123