Searched refs:width_cum_seq (Results 1 – 1 of 1) sorted by relevance
98 std::vector<int64> width_cum_seq; in Compute() local103 width_cum_seq = GeneratePoolingSequence(input_size[2], output_size[2], in Compute()122 2, TensorShape({static_cast<int64>(width_cum_seq.size())}), in Compute()142 for (int i = 0; i < width_cum_seq.size(); ++i) { in Compute()143 output_width_seq_flat(i) = width_cum_seq[i]; in Compute()163 for (int64 ws = 0; ws < width_cum_seq.size() - 1; ++ws) { in Compute()167 const int64 width_start = width_cum_seq[ws]; in Compute()169 overlapping_ ? width_cum_seq[ws + 1] : width_cum_seq[ws + 1] - 1; in Compute()