Searched refs:batch_pos (Results 1 – 10 of 10) sorted by relevance
77 size_t batch_pos = 0; // < kFloatsPerBatch in RandomImage() local79 BitsToFloat(reinterpret_cast<const uint32_t*>(batch) + batch_pos, in RandomImage()81 batch_pos += N; in RandomImage()
36 self.batch_pos = 0 # the starting position of the current mini-batch48 self.batch_pos = 0253 n_start = self.batch_pos258 self.batch_pos = 0261 self.batch_pos = n_end
59 size_t batch_pos; member
98 self->in.batch_pos = 0;1100 sz = self->in.batch_len - self->in.batch_pos;1104 memcpy (buf, self->in.batch + self->in.batch_pos, sz);1105 self->in.batch_pos += sz;1198 self->in.batch_pos = 0;1203 self->in.batch_pos += sz;
80 size_t batch_pos = 0; // < kFloatsPerBatch in RandomImage() local82 BitsToFloat(reinterpret_cast<const uint32_t*>(batch) + batch_pos, in RandomImage()84 batch_pos += N; in RandomImage()
130 group.batch_pos[j].push_back(pos); in assign_mixed_or_batch_idx_helper()161 std::array<dim_vector,N> batch_pos; member249 idx[I][group.batch_pos[j][k]] = A.index(i, group.batch_idx[j][k]); in set_batch_indices_helper()250 stride[I][group.batch_pos[j][k]] = group.batch_stride[group.batch_pos[j][k]]; in set_batch_indices_helper()
158 group.batch_pos[I].push_back(pos); in assign_mixed_or_batch_idx_helper()199 std::array<dim_vector,N> batch_pos; member
209 n_start = self.batch_pos214 self.batch_pos = 0217 self.batch_pos = n_end