Searched refs:outside_to_inside (Results 1 – 1 of 1) sorted by relevance
2967 absl::flat_hash_map<const HloInstruction*, HloInstruction*> outside_to_inside; in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local2970 if (outside_to_inside.count(u) == 0 && to_sink.count(u) > 0 && in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()2972 return outside_to_inside.count(o) > 0; in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()2979 outside_to_inside[new_operands[i]] = in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()3005 inst_new_operands[i] = outside_to_inside[inst->operand(i)]; in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()3007 outside_to_inside[inst] = body->AddInstruction( in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()3011 TF_RET_CHECK(outside_to_inside.count(old_operand) > 0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()3016 ou->ReplaceAllUsesWith(outside_to_inside[old_operand])); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()3194 absl::flat_hash_map<const HloInstruction*, HloInstruction*> outside_to_inside; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3198 if (outside_to_inside.count(u) == 0 && to_move.count(u) > 0 && in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()[all …]