Searched refs:window_bound (Results 1 – 2 of 2) sorted by relevance
127 int64_t window_bound; in XlaGather() local130 window_bound = 1; in XlaGather()132 window_bound = input_shape.dim_size(i); in XlaGather()135 slice_sizes.push_back(window_bound); in XlaGather()
154 int64_t window_bound; in GetMatrixDiagonalViaGather() local158 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()