Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.cc64 MemoryTypeVector out_mvec; in ProcessMemoryTypes() local
67 n->def(), &inp_mvec, &out_mvec)); in ProcessMemoryTypes()
72 for (size_t i = 0; i < out_mvec.size(); ++i) { in ProcessMemoryTypes()
73 VLOG(2) << "out mvec " << n->id() << " " << i << " " << out_mvec[i]; in ProcessMemoryTypes()
74 out[{n->id(), static_cast<int>(i)}] = out_mvec[i]; in ProcessMemoryTypes()
208 MemoryTypeVector out_mvec; in MemoryTypeForOutput() local
210 &inp_mvec, &out_mvec)); in MemoryTypeForOutput()
211 if (out_mvec.size() <= index) { in MemoryTypeForOutput()
214 " that has only ", out_mvec.size(), " outputs"); in MemoryTypeForOutput()
216 *memory_type = out_mvec[index]; in MemoryTypeForOutput()