Home
last modified time | relevance | path

Searched refs:src_it (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.cc168 auto src_it = buffers().find(index); in TakeSubTree() local
171 dst_it->second = src_it->second; in TakeSubTree()
172 src_it->second = tensorflow::se::DeviceMemoryBase(nullptr, 0); in TakeSubTree()
173 ++src_it; in TakeSubTree()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc135 auto src_it = info.output_types.find({src->id(), src_port}); in NeedSameDeviceSendRecv() local
136 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() local
205 DCHECK(src_it != g_info.output_types.end()); in AddSend()
206 host_memory = (src_it->second == HOST_MEMORY); in AddSend()