Searched refs:slices_padded (Results 1 – 1 of 1) sorted by relevance
126 std::array<XlaOp, kNumSlices> slices_padded; in PrefixSum() local133 slices_padded[i] = in PrefixSum()137 XlaOp ret = slices_padded[0]; in PrefixSum()139 ret = ret + slices_padded[i]; in PrefixSum()