Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc337 int64 i3_base = i3 * stride[3] - pad_low[3]; 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()
425 int64 i3_base = i3 * stride[3] - pad_low[3]; in SelectAndScatter4DGePlus() local
429 int64 scatter_3 = (i3_base >= 0) ? i3_base : 0; in SelectAndScatter4DGePlus()
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()