Searched refs:batch_shift (Results 1 – 4 of 4) sorted by relevance
| /dports/devel/xsimd/xsimd-7.6.0/test/ |
| H A D | test_batch_int.cpp | 232 batch_type res = batch_lhs() << batch_shift(); in test_shift()
248 batch_type res = batch_lhs() >> batch_shift(); in test_shift()
311 batch_type batch_shift() const in batch_shift() function in batch_int_test
|
| /dports/misc/py-gluonnlp/gluon-nlp-0.10.0/src/gluonnlp/model/ |
| H A D | sequence_sampler.py | 277 states, vocab_size, batch_shift): argument
347 batch_beam_indices = F.broadcast_add(beam_ids, F.expand_dims(batch_shift, axis=1))
652 batch_shift = F.arange(0, batch_size * beam_size, beam_size, dtype=np.int32)
668 vocab_size, batch_shift)
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
| H A D | ordering_op-inl.h | 844 Tensor<xpu, 1, index_t> batch_shift = in TopKBackwardImpl() local
852 batch_shift.dptr_); in TopKBackwardImpl()
857 broadcast_to(inplace_reshape(batch_shift, in TopKBackwardImpl()
869 broadcast_to(inplace_reshape(batch_shift, Shape2(batch_size, 1)), in TopKBackwardImpl()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
| H A D | ordering_op-inl.h | 844 Tensor<xpu, 1, index_t> batch_shift = in TopKBackwardImpl() local
852 batch_shift.dptr_); in TopKBackwardImpl()
857 broadcast_to(inplace_reshape(batch_shift, in TopKBackwardImpl()
869 broadcast_to(inplace_reshape(batch_shift, Shape2(batch_size, 1)), in TopKBackwardImpl()
|