Searched refs:out_mtypes (Results 1 – 3 of 3) sorted by relevance
78 MemoryTypeVector* out_mtypes) { in MemoryTypesForNode() argument94 out_mtypes->clear(); in MemoryTypesForNode()104 for (const auto& t : out_dtypes) out_mtypes->push_back(MTypeFromDType(t)); in MemoryTypesForNode()115 out_mtypes->resize(GetTotal(out_names), DEVICE_MEMORY); in MemoryTypesForNode()121 MemoryTypesHelper(out_names, &host_memory_args, out_mtypes); in MemoryTypesForNode()128 CHECK_LE(out_mtypes->size(), out_dtypes.size()); in MemoryTypesForNode()136 for (int i = 0; i < out_mtypes->size(); ++i) { in MemoryTypesForNode()138 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()152 if (0 <= i && i < out_mtypes->size()) { in MemoryTypesForNode()153 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
257 tensorflow::MemoryTypeVector out_mtypes; in TF_GetSupportedDevices() local259 &inp_mtypes, &out_mtypes); in TF_GetSupportedDevices()
380 tensorflow::MemoryTypeVector out_mtypes; in IsHostMemory() local382 node, &in_mtypes, &out_mtypes); in IsHostMemory()384 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()