Searched refs:UpdateDDNums (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 113 void UpdateDDNums(DotDimensionNumbers* new_ddnums, int64 reshaped_dim, in UpdateDDNums() function
868 UpdateDDNums(&new_ddnums, slice_sharding_dim, true); in PartitionBaseCase()
874 UpdateDDNums(&new_ddnums, slice_sharding_dim, false); in PartitionBaseCase()
903 UpdateDDNums(&new_ddnums, lhs_concat_dim, true); in PartitionBaseCase()
926 UpdateDDNums(&new_ddnums, rhs_concat_dim, false); in PartitionBaseCase()
|