Searched refs:i2_base (Results 1 – 1 of 1) sorted by relevance
274 int64 i2_base = i2 * stride[2] - pad_low[2]; in ReduceWindow3DAdd() local281 i2_base + i2_win >= 0 && i0_base + i0_win < operand.n1() && in ReduceWindow3DAdd()283 i2_base + i2_win < operand.n3()) { in ReduceWindow3DAdd()285 i2_base + i2_win); in ReduceWindow3DAdd()336 int64 i2_base = i2 * stride[2] - pad_low[2]; in ReduceWindow4DGeneric() local345 i2_base + i2_win >= 0 && i3_base + i3_win >= 0 && in ReduceWindow4DGeneric()348 i2_base + i2_win < operand.n3() && in ReduceWindow4DGeneric()352 i2_base + i2_win, i3_base + i3_win)); in ReduceWindow4DGeneric()424 int64 i2_base = i2 * stride[2] - pad_low[2]; in SelectAndScatter4DGePlus() local428 int64 scatter_2 = (i2_base >= 0) ? i2_base : 0; in SelectAndScatter4DGePlus()[all …]