Searched refs:all_reduce_start_ops_ (Results 1 – 2 of 2) sorted by relevance
1209 TF_RET_CHECK(all_reduce_start_ops_.emplace(instr, all_reduce_start_op).second) in EmitAllReduceStartOp()1216 auto it = all_reduce_start_ops_.find(instr->operand(0)); in EmitAllReduceDoneOp()1217 TF_RET_CHECK(it != all_reduce_start_ops_.end()) in EmitAllReduceDoneOp()1222 all_reduce_start_ops_.erase(it); in EmitAllReduceDoneOp()
269 all_reduce_start_ops_; variable