Searched refs:new_outputs_inside (Results 1 – 1 of 1) sorted by relevance
3284 std::vector<HloInstruction*> new_outputs_inside(new_operands.size()); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3285 for (int64 i = 0; i < new_outputs_inside.size(); ++i) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3286 new_outputs_inside[i] = outside_to_inside[new_operands[i]]; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3365 new_outputs_inside[index_in_operand] = output_inside; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3369 body->AddInstruction(HloInstruction::CreateTuple(new_outputs_inside)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3381 new_outputs_inside[index_in_operand]->shape(), new_output_gte, in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()