Searched refs:NeedSameDeviceSendRecv (Results 1 – 1 of 1) sorted by relevance
124 bool NeedSameDeviceSendRecv(const Edge* edge, const GraphInfo& info) { in NeedSameDeviceSendRecv() function211 if (dtype != cast_dtype && !NeedSameDeviceSendRecv(edge, g_info)) { in AddSend()260 if (opts.should_cast && !NeedSameDeviceSendRecv(edge, g_info)) { in AddRecv()1046 if (src_graph == dst_graph && !NeedSameDeviceSendRecv(edge, g_info)) { in Partition()