File tree Expand file tree Collapse file tree 1 file changed +0
-9
lines changed Expand file tree Collapse file tree 1 file changed +0
-9
lines changed Original file line number Diff line number Diff line change @@ -53,13 +53,8 @@ class SamplingIdKernel : public framework::OpKernel<T> {
53
53
static_cast <T>(context.Attr <float >(" min" )),
54
54
static_cast <T>(context.Attr <float >(" max" )));
55
55
56
- <<<<<<< HEAD
57
56
std::vector<int64_t > ids (batch_size);
58
57
for (size_t i = 0 ; i < batch_size; ++i) {
59
- =======
60
- std::vector<T> ids (batch_size);
61
- for (int i = 0 ; i < batch_size; ++i) {
62
- >>>>>>> 823c4f87beff04e4029e3f4a183658621ca8f01b
63
58
T r = dist (engine);
64
59
int idx = width - 1 ;
65
60
for (int j = 0 ; j < width; ++j) {
@@ -68,11 +63,7 @@ class SamplingIdKernel : public framework::OpKernel<T> {
68
63
break ;
69
64
}
70
65
}
71
- <<<<<<< HEAD
72
66
ids[i] = int64_t (idx);
73
- =======
74
- ids[i] = ins_vector[idx];
75
- >>>>>>> 823c4f87beff04e4029e3f4a183658621ca8f01b
76
67
}
77
68
78
69
std::vector<int64_t > out_dim;
You can’t perform that action at this time.
0 commit comments