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.cc212 for (int64 i1_win = 0; i1_win < window[1]; ++i1_win) { in ReduceWindow3DAdd() local
214 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow3DAdd()
216 i1_base + i1_win < operand.n2() && in ReduceWindow3DAdd()
218 val += operand(i0_base + i0_win, i1_base + i1_win, in ReduceWindow3DAdd()
275 for (int64 i1_win = 0; i1_win < window[1]; ++i1_win) { in ReduceWindow4DGeneric() local
278 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow4DGeneric()
281 i1_base + i1_win < operand.n2() && in ReduceWindow4DGeneric()
285 val, operand(i0_base + i0_win, i1_base + i1_win, in ReduceWindow4DGeneric()
366 for (int64 i1_win = 0; i1_win < window[1]; ++i1_win) { in SelectAndScatter4DGePlus() local
369 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in SelectAndScatter4DGePlus()
[all …]