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