Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc343 for (int64 i3_win = 0; i3_win < window[3]; ++i3_win) { in ReduceWindow4DGeneric() local
345 i2_base + i2_win >= 0 && i3_base + i3_win >= 0 && in ReduceWindow4DGeneric()
349 i3_base + i3_win < operand.n4()) { in ReduceWindow4DGeneric()
352 i2_base + i2_win, i3_base + i3_win)); in ReduceWindow4DGeneric()
434 for (int64 i3_win = 0; i3_win < window[3]; ++i3_win) { in SelectAndScatter4DGePlus() local
436 i2_base + i2_win >= 0 && i3_base + i3_win >= 0 && in SelectAndScatter4DGePlus()
440 i3_base + i3_win < operand.n4()) { in SelectAndScatter4DGePlus()
442 i2_base + i2_win, i3_base + i3_win); in SelectAndScatter4DGePlus()
448 scatter_3 = i3_base + i3_win; in SelectAndScatter4DGePlus()