Searched refs:dims_to_mask (Results 1 – 1 of 1) sorted by relevance
3307 std::vector<int64> dims_to_mask; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3314 dims_to_mask.push_back(i); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3322 for (int64 dim : dims_to_mask) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()