Searched refs:i0_base (Results 1 – 1 of 1) sorted by relevance
177 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow1DGeneric() local181 if (i0_base + i0_win >= 0 && i0_base + i0_win < dim_lengths[0]) { in ReduceWindow1DGeneric()182 val = reduce_func(val, operand[i0_base + i0_win]); in ReduceWindow1DGeneric()224 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow2DGeneric() local230 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow2DGeneric()231 i0_base + i0_win < operand.n1() && in ReduceWindow2DGeneric()233 val = reduce_func(val, operand(i0_base + i0_win, i1_base + i1_win)); in ReduceWindow2DGeneric()272 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow3DAdd() local280 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow3DAdd()281 i2_base + i2_win >= 0 && i0_base + i0_win < operand.n1() && in ReduceWindow3DAdd()[all …]