Searched refs:slice_limit (Results 1 – 2 of 2) sorted by relevance
173 std::vector<int64> slice_limit(result_rank); in Compile() local176 slice_limit[1 + d - indices0_shape.dims()] = data0_shape.dim_size(d); in Compile()183 slice_limit[0] = src_slice_vector[index_num] + 1; in Compile()187 xla::Slice(expression, slice_start, slice_limit, stride); in Compile()
206 int64 slice_limit; member225 Slice(original, {spec.slice_start}, {spec.slice_limit}, in Run()230 for (int i = spec.slice_start; i < spec.slice_limit; in Run()247 spec.slice_limit, spec.slice_stride); in SliceR1TestDataToString()