Searched refs:i0_base (Results 1 – 1 of 1) sorted by relevance
163 int64 i0_base = i0 * stride_amount - pad_low; in ReduceWindow1DGeneric() local166 if (i0_base + i0_win >= 0 && i0_base + i0_win < operand.size()) { in ReduceWindow1DGeneric()167 val = reduce_func(val, operand[i0_base + i0_win]); in ReduceWindow1DGeneric()206 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow3DAdd() local214 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow3DAdd()215 i2_base + i2_win >= 0 && i0_base + i0_win < operand.n1() && in ReduceWindow3DAdd()218 val += operand(i0_base + i0_win, i1_base + i1_win, in ReduceWindow3DAdd()268 int64 i0_base = i0 * stride[0] - pad_low[0]; in ReduceWindow4DGeneric() local278 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow4DGeneric()280 i0_base + i0_win < operand.n1() && in ReduceWindow4DGeneric()[all …]