Searched refs:i2_base (Results 1 – 1 of 1) sorted by relevance
208 int64 i2_base = i2 * stride[2] - pad_low[2]; in ReduceWindow3DAdd() local215 i2_base + i2_win >= 0 && i0_base + i0_win < operand.n1() && in ReduceWindow3DAdd()217 i2_base + i2_win < operand.n3()) { in ReduceWindow3DAdd()219 i2_base + i2_win); in ReduceWindow3DAdd()270 int64 i2_base = i2 * stride[2] - pad_low[2]; in ReduceWindow4DGeneric() local279 i2_base + i2_win >= 0 && i3_base + i3_win >= 0 && in ReduceWindow4DGeneric()282 i2_base + i2_win < operand.n3() && in ReduceWindow4DGeneric()286 i2_base + i2_win, i3_base + i3_win)); in ReduceWindow4DGeneric()358 int64 i2_base = i2 * stride[2] - pad_low[2]; in SelectAndScatter4DGePlus() local362 int64 scatter_2 = (i2_base >= 0) ? i2_base : 0; in SelectAndScatter4DGePlus()[all …]