Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc229 for (int64 i1_win = 0; i1_win < window[1]; ++i1_win) { in ReduceWindow2DGeneric() local
230 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow2DGeneric()
232 i1_base + i1_win < operand.n2()) { in ReduceWindow2DGeneric()
233 val = reduce_func(val, operand(i0_base + i0_win, i1_base + i1_win)); in ReduceWindow2DGeneric()
278 for (int64 i1_win = 0; i1_win < window[1]; ++i1_win) { in ReduceWindow3DAdd() local
280 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow3DAdd()
282 i1_base + i1_win < operand.n2() && in ReduceWindow3DAdd()
284 val += operand(i0_base + i0_win, i1_base + i1_win, in ReduceWindow3DAdd()
341 for (int64 i1_win = 0; i1_win < window[1]; ++i1_win) { in ReduceWindow4DGeneric() local
344 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow4DGeneric()
[all …]