Searched refs:cw_cp_output (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()
1028 cw_cp_output = body_b.AddInstruction(HloInstruction::CreateBinary( in PartitionBaseCase()
1029 o->shape(), HloOpcode::kAdd, cw_cp_output, cw_dot)); in PartitionBaseCase()
1050 partial_results.push_back(cw_cp_output); in PartitionBaseCase()
1177 auto cw_cp_output = in PartitionBaseCase() local
1185 get_partial_bid_results(l, r, o, extra_inout, cw_cp_output, i)); in PartitionBaseCase()
1187 cw_cp_output = outputs[1]; in PartitionBaseCase()
1211 auto next_cw_cp_input = cw_cp_output; in PartitionBaseCase()
1219 outputs, get_partial_bid_results(next_l, next_r, o, cw_cp_output, in PartitionBaseCase()
|