Searched refs:col_cum_seq (Results 1 – 1 of 1) sorted by relevance
93 std::vector<int64> col_cum_seq; in Compute() local98 col_cum_seq = GeneratePoolingSequence(input_size[2], output_size[2], in Compute()116 2, TensorShape({static_cast<int64>(col_cum_seq.size())}), in Compute()139 for (int i = 0; i < col_cum_seq.size(); ++i) { in Compute()140 output_col_seq_flat(i) = col_cum_seq[i]; in Compute()160 for (int64 ws = 0; ws < col_cum_seq.size() - 1; ++ws) { in Compute()164 const int64 col_start = col_cum_seq[ws]; in Compute()166 overlapping_ ? col_cum_seq[ws + 1] : col_cum_seq[ws + 1] - 1; in Compute()