Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc127 int64_t window_bound; in XlaGather() local
130 window_bound = 1; in XlaGather()
132 window_bound = input_shape.dim_size(i); in XlaGather()
135 slice_sizes.push_back(window_bound); in XlaGather()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc154 int64_t window_bound; in GetMatrixDiagonalViaGather() local
158 window_bound = (shape.dimensions(i) != 0) ? 1 : 0; in GetMatrixDiagonalViaGather()
161 window_bound = shape.dimensions(i); in GetMatrixDiagonalViaGather()
163 slice_sizes.push_back(window_bound); in GetMatrixDiagonalViaGather()