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.cc279 for (int64 i2_win = 0; i2_win < window[2]; ++i2_win) { in ReduceWindow3DAdd() local
281 i2_base + i2_win >= 0 && i0_base + i0_win < operand.n1() && in ReduceWindow3DAdd()
283 i2_base + i2_win < operand.n3()) { in ReduceWindow3DAdd()
285 i2_base + i2_win); in ReduceWindow3DAdd()
342 for (int64 i2_win = 0; i2_win < window[2]; ++i2_win) { in ReduceWindow4DGeneric() local
345 i2_base + i2_win >= 0 && i3_base + i3_win >= 0 && in ReduceWindow4DGeneric()
348 i2_base + i2_win < operand.n3() && in ReduceWindow4DGeneric()
352 i2_base + i2_win, i3_base + i3_win)); in ReduceWindow4DGeneric()
433 for (int64 i2_win = 0; i2_win < window[2]; ++i2_win) { in SelectAndScatter4DGePlus() local
436 i2_base + i2_win >= 0 && i3_base + i3_win >= 0 && in SelectAndScatter4DGePlus()
[all …]