Searched refs:extra_inout (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 664 HloInstruction* extra_inout, HloInstruction* cw_cp_output, in PartitionBaseCase()
881 auto cw_dot_lhs = windowed_op_is_lhs ? extra_inout : l; in PartitionBaseCase()
882 auto cw_dot_rhs = windowed_op_is_lhs ? r : extra_inout; in PartitionBaseCase()
1138 auto extra_inout = body_b.AddInstruction( in PartitionBaseCase() local
1176 auto cw_cp_input = extra_inout; in PartitionBaseCase()
1185 get_partial_bid_results(l, r, o, extra_inout, cw_cp_output, i)); in PartitionBaseCase()
1246 TF_ASSIGN_OR_RETURN(extra_inout, in PartitionBaseCase()
1247 get_partial_unid_result(l, r, extra_inout, i)); in PartitionBaseCase()
1249 extra_inout = lhs.state() in PartitionBaseCase()
1252 &body_b, extra_inout, output_sd_pairs, in PartitionBaseCase()
[all …]
|