Searched refs:window_bound (Results 1 – 2 of 2) sorted by relevance
126 int64 window_bound; in XlaGather() local129 window_bound = 1; in XlaGather()131 window_bound = input_shape.dim_size(i); in XlaGather()134 slice_sizes.push_back(window_bound); in XlaGather()
153 int64 window_bound; in GetMatrixDiagonalViaGather() local157 window_bound = (shape.dimensions(i) != 0) ? 1 : 0; in GetMatrixDiagonalViaGather()160 window_bound = shape.dimensions(i); in GetMatrixDiagonalViaGather()162 slice_sizes.push_back(window_bound); in GetMatrixDiagonalViaGather()