Home
last modified time | relevance | path

Searched refs:i0_win (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc180 for (int64 i0_win = 0; i0_win < window[0]; ++i0_win) { in ReduceWindow1DGeneric() local
181 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()
228 for (int64 i0_win = 0; i0_win < window[0]; ++i0_win) { in ReduceWindow2DGeneric() local
230 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()
277 for (int64 i0_win = 0; i0_win < window[0]; ++i0_win) { in ReduceWindow3DAdd() local
280 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 …]