Home
last modified time | relevance | path

Searched refs:add_users_if_available (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2968 auto add_users_if_available = [&](HloInstruction* inst) { in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
2982 add_users_if_available(new_operands[i]); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3009 add_users_if_available(inst); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3196 auto add_users_if_available = [&](HloInstruction* inst) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
3210 add_users_if_available(new_operands[i]); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3217 add_users_if_available(original_output); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3282 add_users_if_available(inst); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()