Searched refs:reduce_outputs (Results 1 – 1 of 1) sorted by relevance
3060 std::vector<HloInstruction*> reduce_outputs; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3093 reduce_outputs.push_back(inst); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3146 for (auto out : reduce_outputs) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3289 for (int64 i = 0; i < reduce_outputs.size(); ++i) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3290 auto reduce_outside = reduce_outputs[i]; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3292 int64 index_in_operand = new_operands.size() - reduce_outputs.size() + i; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3377 for (int64 i = 0; i < reduce_outputs.size(); ++i) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3378 int64 index_in_operand = new_operands.size() - reduce_outputs.size() + i; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3384 reduce_outputs[i]->shape())) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3386 reduce_outputs[i]->shape(), new_output, in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()[all …]