Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc207 int64 i1_base = i1 * stride[1] - pad_low[1]; 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()
269 int64 i1_base = i1 * stride[1] - pad_low[1]; 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()
357 int64 i1_base = i1 * stride[1] - pad_low[1]; in SelectAndScatter4DGePlus() local
361 int64 scatter_1 = (i1_base >= 0) ? i1_base : 0; in SelectAndScatter4DGePlus()
[all …]