Searched refs:height_cum_seq (Results 1 – 1 of 1) sorted by relevance
97 std::vector<int64> height_cum_seq; in Compute() local101 height_cum_seq = GeneratePoolingSequence(input_size[1], output_size[1], in Compute()117 1, TensorShape({static_cast<int64>(height_cum_seq.size())}), in Compute()138 for (int i = 0; i < height_cum_seq.size(); ++i) { in Compute()139 output_height_seq_flat(i) = height_cum_seq[i]; in Compute()155 for (int64 hs = 0; hs < height_cum_seq.size() - 1; ++hs) { in Compute()157 const int64 height_start = height_cum_seq[hs]; in Compute()159 overlapping_ ? height_cum_seq[hs + 1] : height_cum_seq[hs + 1] - 1; in Compute()