Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc213 for (int64 i2_win = 0; i2_win < window[2]; ++i2_win) { in ReduceWindow3DAdd() local
215 i2_base + i2_win >= 0 && i0_base + i0_win < operand.n1() && in ReduceWindow3DAdd()
217 i2_base + i2_win < operand.n3()) { in ReduceWindow3DAdd()
219 i2_base + i2_win); in ReduceWindow3DAdd()
276 for (int64 i2_win = 0; i2_win < window[2]; ++i2_win) { in ReduceWindow4DGeneric() local
279 i2_base + i2_win >= 0 && i3_base + i3_win >= 0 && in ReduceWindow4DGeneric()
282 i2_base + i2_win < operand.n3() && in ReduceWindow4DGeneric()
286 i2_base + i2_win, i3_base + i3_win)); in ReduceWindow4DGeneric()
367 for (int64 i2_win = 0; i2_win < window[2]; ++i2_win) { in SelectAndScatter4DGePlus() local
370 i2_base + i2_win >= 0 && i3_base + i3_win >= 0 && in SelectAndScatter4DGePlus()
[all …]