Searched refs:reduce_dus_offsets (Results 1 – 1 of 1) sorted by relevance
3302 std::vector<HloInstruction*> reduce_dus_offsets; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3319 reduce_dus_offsets.push_back(slice_offsets[i]); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3350 output_inside->shape(), last_iter_result, reduce_dus_offsets, in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3363 reduce_dus_offsets)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()