Searched refs:src_it (Results 1 – 2 of 2) sorted by relevance
168 auto src_it = buffers().find(index); in TakeSubTree() local171 dst_it->second = src_it->second; in TakeSubTree()172 src_it->second = tensorflow::se::DeviceMemoryBase(nullptr, 0); in TakeSubTree()173 ++src_it; in TakeSubTree()
135 auto src_it = info.output_types.find({src->id(), src_port}); in NeedSameDeviceSendRecv() local136 DCHECK(src_it != info.output_types.end()); in NeedSameDeviceSendRecv()139 return src_it->second != dst_it->second; in NeedSameDeviceSendRecv()204 auto src_it = g_info.output_types.find({src->id(), src_port}); in AddSend() local205 DCHECK(src_it != g_info.output_types.end()); in AddSend()206 host_memory = (src_it->second == HOST_MEMORY); in AddSend()