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.cc225 int64 i1_base = i1 * stride[1] - pad_low[1]; in ReduceWindow2DGeneric() local
230 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow2DGeneric()
232 i1_base + i1_win < operand.n2()) { in ReduceWindow2DGeneric()
233 val = reduce_func(val, operand(i0_base + i0_win, i1_base + i1_win)); in ReduceWindow2DGeneric()
273 int64 i1_base = i1 * stride[1] - pad_low[1]; in ReduceWindow3DAdd() local
280 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow3DAdd()
282 i1_base + i1_win < operand.n2() && in ReduceWindow3DAdd()
284 val += operand(i0_base + i0_win, i1_base + i1_win, in ReduceWindow3DAdd()
335 int64 i1_base = i1 * stride[1] - pad_low[1]; in ReduceWindow4DGeneric() local
344 if (i0_base + i0_win >= 0 && i1_base + i1_win >= 0 && in ReduceWindow4DGeneric()
[all …]