Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc794 auto lhs_sharding_transposed_to_match_output = in PartitionBaseCase() local
818 dot->set_sharding(*lhs_sharding_transposed_to_match_output); in PartitionBaseCase()
833 lhs_sharding_transposed_to_match_output == output_sharding) { in PartitionBaseCase()
904 windowed_op_is_lhs ? *lhs_sharding_transposed_to_match_output in PartitionBaseCase()
943 ? &*lhs_sharding_transposed_to_match_output in PartitionBaseCase()
1394 windowed_op_is_lhs ? *lhs_sharding_transposed_to_match_output in PartitionBaseCase()
1399 windowed_op_is_lhs ? *lhs_sharding_transposed_to_match_output in PartitionBaseCase()
1468 windowed_op_is_lhs ? *lhs_sharding_transposed_to_match_output in PartitionBaseCase()
1891 lhs_sharding_transposed_to_match_output == output_sharding) { in PartitionBaseCase()