Searched refs:get_partial_unid_result (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1055 auto get_partial_unid_result = in PartitionBaseCase() local
1247 get_partial_unid_result(l, r, extra_inout, i)); in PartitionBaseCase()
1265 TF_ASSIGN_OR_RETURN(o, get_partial_unid_result(l, r, o, real_i)); in PartitionBaseCase()
1290 TF_ASSIGN_OR_RETURN(o, get_partial_unid_result(l, r, o, i)); in PartitionBaseCase()
1312 TF_ASSIGN_OR_RETURN(o, get_partial_unid_result(next_l, next_r, o, i)); in PartitionBaseCase()
1333 TF_ASSIGN_OR_RETURN(o, get_partial_unid_result(l, r, o, real_i)); in PartitionBaseCase()
|