Searched refs:g_info (Results 1 – 1 of 1) sorted by relevance
192 NodeDef* AddSend(const PartitionOptions& opts, const GraphInfo& g_info, in AddSend() argument204 auto src_it = g_info.output_types.find({src->id(), src_port}); in AddSend()205 DCHECK(src_it != g_info.output_types.end()); in AddSend()211 if (dtype != cast_dtype && !NeedSameDeviceSendRecv(edge, g_info)) { in AddSend()250 NodeDef* AddRecv(const PartitionOptions& opts, const GraphInfo& g_info, in AddRecv() argument260 if (opts.should_cast && !NeedSameDeviceSendRecv(edge, g_info)) { in AddRecv()267 auto dst_it = g_info.input_types.find({dst->id(), dst_port}); in AddRecv()268 DCHECK(dst_it != g_info.input_types.end()); in AddRecv()609 GraphInfo* g_info) { in AddControlFlow() argument612 std::vector<ControlFlowInfo>& cf_info = g_info->cf_info; in AddControlFlow()[all …]