Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc118 auto dst_it = attr.find("_dst"); in TraceString() local
120 const string& dst = dst_it != attr.end() ? dst_it->second.s() : ""; in TraceString()
162 auto dst_it = attr.find("_dst"); in TraceString() local
164 const string& dst = dst_it != attr.end() ? dst_it->second.s() : ""; in TraceString()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.cc182 auto dst_it = output.buffers().begin(); in TakeSubTree() local
183 while (dst_it != output.buffers().end()) { in TakeSubTree()
184 dst_it->second = src_it->second; in TakeSubTree()
187 ++dst_it; in TakeSubTree()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc139 auto dst_it = info.input_types.find({dst->id(), dst_port}); in NeedSameDeviceSendRecv() local
140 DCHECK(dst_it != info.input_types.end()); in NeedSameDeviceSendRecv()
141 return src_it->second != dst_it->second; in NeedSameDeviceSendRecv()
153 auto dst_it = info.input_types.find({dst->id(), dst_port}); in IsDstInputOnHost() local
154 DCHECK(dst_it != info.input_types.end()); in IsDstInputOnHost()
155 return dst_it->second == HOST_MEMORY; in IsDstInputOnHost()
272 auto dst_it = g_info.input_types.find({dst->id(), dst_port}); in AddRecv() local
273 DCHECK(dst_it != g_info.input_types.end()); in AddRecv()
274 host_memory = (dst_it->second == HOST_MEMORY); in AddRecv()