Home
last modified time | relevance | path

Searched defs:resharded_lhs (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_test.cc7572 auto resharded_lhs = in TEST_F() local
8425 auto resharded_lhs = AllOf( in TEST_F() local
8499 auto resharded_lhs = AllOf( in TEST_F() local
8650 auto resharded_lhs = AllOf( in TEST_F() local
8801 auto resharded_lhs = in TEST_F() local
8900 auto resharded_lhs = AllOf( in TEST_F() local
9155 auto resharded_lhs = in TEST_F() local
H A Ddot_handler.cc858 auto resharded_lhs = lhs.Reshard(*rhs_sharding_transposed_to_match_lhs); in PartitionBaseCase() local
1911 auto resharded_lhs = in PartitionBaseCase() local
1922 auto resharded_lhs = in PartitionBaseCase() local