Home | History | Annotate | Download | only in kernels

Lines Matching defs:diff

23   std::vector<int64> diff(output_length, 0);
75 diff[i] = cum_seq[i + 1] - cum_seq[i];
78 return diff;
85 std::vector<int64> diff(output_length, k);
88 diff[i] += 1;
92 auto local_gen = generator->ReserveSamples32(diff.size());
95 RandomShuffle(diff.begin(), diff.end(), uniform);
97 return diff;
103 std::vector<int64> diff;
104 // This is a case that regular pooling can handle, just return diff with
107 diff = std::vector<int64>(output_length, input_length / output_length);
111 diff = GeneratePoolingSequencePseudoRandom(input_length, output_length,
114 diff =
121 // k<= diff[i] <= k+1.
122 DCHECK_GE(diff[i], k);
123 DCHECK_LE(diff[i], k + 1);
129 cum_seq[i] = cum_seq[i - 1] + diff[i - 1];