Home
last modified time | relevance | path

Searched refs:cum_seq (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfractional_pool_common.cc22 std::vector<int64> cum_seq(output_length + 1, 0); in GeneratePoolingSequencePseudoRandom() local
68 cum_seq[0] = 1; in GeneratePoolingSequencePseudoRandom()
69 cum_seq[output_length] = input_length + 1; in GeneratePoolingSequencePseudoRandom()
71 cum_seq[i] = static_cast<int>(ceil(alpha * (i + u))); in GeneratePoolingSequencePseudoRandom()
75 diff[i] = cum_seq[i + 1] - cum_seq[i]; in GeneratePoolingSequencePseudoRandom()
127 std::vector<int64> cum_seq(output_length + 1, 0); in GeneratePoolingSequence() local
128 for (int i = 1; i < cum_seq.size(); ++i) { in GeneratePoolingSequence()
129 cum_seq[i] = cum_seq[i - 1] + diff[i - 1]; in GeneratePoolingSequence()
131 return cum_seq; in GeneratePoolingSequence()