Fix call to HloDataflowAnalysis::CanShareOperandBufferWithUser inside of HloOrdering::UsesBeforeValueDefinition.
Fix call to HloDataflowAnalysis::CanShareOperandBufferWithUser inside of HloOrdering::UsesBeforeValueDefinition.

The current invocation does not consider the case in which the value instruction is an async wrapped call instruction residing in a separate computation.

Added unit test in hlo_ordering_test.

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/12537 from Intel-tensorflow:sachin/softmax-axis de9d6bd36272962faf76077f594649d229bb727a
