Searched refs:row_cum_seq (Results 1 – 1 of 1) sorted by relevance
92 std::vector<int64> row_cum_seq; in Compute() local96 row_cum_seq = GeneratePoolingSequence(input_size[1], output_size[1], in Compute()111 1, TensorShape({static_cast<int64>(row_cum_seq.size())}), in Compute()135 for (int i = 0; i < row_cum_seq.size(); ++i) { in Compute()136 output_row_seq_flat(i) = row_cum_seq[i]; in Compute()152 for (int64 hs = 0; hs < row_cum_seq.size() - 1; ++hs) { in Compute()154 const int64 row_start = row_cum_seq[hs]; in Compute()156 overlapping_ ? row_cum_seq[hs + 1] : row_cum_seq[hs + 1] - 1; in Compute()