Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc2793 std::vector<int64> new_slice_limits(new_slice_shape.dimensions().begin(), in TryToReorderSliceAndReshape() local
2797 if (slice_elements >= new_slice_limits[i]) { in TryToReorderSliceAndReshape()
2798 if (slice_elements % new_slice_limits[i] != 0) { in TryToReorderSliceAndReshape()
2801 slice_elements /= new_slice_limits[i]; in TryToReorderSliceAndReshape()
2803 new_slice_limits[i] = slice_elements; in TryToReorderSliceAndReshape()
2810 new_slice_limits), in TryToReorderSliceAndReshape()
2811 new_slice_operand, new_slice_starts, new_slice_limits, in TryToReorderSliceAndReshape()
2830 std::vector<int64> new_slice_limits = slice->slice_limits(); in HandleSlice() local
2833 new_slice_limits[i] += operand_slice->slice_starts(i); in HandleSlice()
2838 new_slice_starts, new_slice_limits, slice->slice_strides())); in HandleSlice()