Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.cc169 auto dst_it = output.buffers().begin(); in TakeSubTree() local
170 while (dst_it != output.buffers().end()) { in TakeSubTree()
171 dst_it->second = src_it->second; in TakeSubTree()
174 ++dst_it; in TakeSubTree()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc137 auto dst_it = info.input_types.find({dst->id(), dst_port}); in NeedSameDeviceSendRecv() local
138 DCHECK(dst_it != info.input_types.end()); in NeedSameDeviceSendRecv()
139 return src_it->second != dst_it->second; in NeedSameDeviceSendRecv()
151 auto dst_it = info.input_types.find({dst->id(), dst_port}); in IsDstInputOnHost() local
152 DCHECK(dst_it != info.input_types.end()); in IsDstInputOnHost()
153 return dst_it->second == HOST_MEMORY; in IsDstInputOnHost()
267 auto dst_it = g_info.input_types.find({dst->id(), dst_port}); in AddRecv() local
268 DCHECK(dst_it != g_info.input_types.end()); in AddRecv()
269 host_memory = (dst_it->second == HOST_MEMORY); in AddRecv()