Searched refs:g_info (Results 1 – 1 of 1) sorted by relevance
196 NodeDef* AddSend(const PartitionOptions& opts, const GraphInfo& g_info, in AddSend() argument208 auto src_it = g_info.output_types.find({src->id(), src_port}); in AddSend()209 DCHECK(src_it != g_info.output_types.end()); in AddSend()215 if (dtype != cast_dtype && !NeedSameDeviceSendRecv(edge, g_info)) { in AddSend()254 NodeDef* AddRecv(const PartitionOptions& opts, const GraphInfo& g_info, in AddRecv() argument264 if (opts.should_cast && !NeedSameDeviceSendRecv(edge, g_info)) { in AddRecv()272 auto dst_it = g_info.input_types.find({dst->id(), dst_port}); in AddRecv()273 DCHECK(dst_it != g_info.input_types.end()); in AddRecv()278 auto src_it = g_info.output_types.find({src->id(), src_port}); in AddRecv()279 DCHECK(src_it != g_info.output_types.end()); in AddRecv()[all …]