Searched refs:inp_mtypes (Results 1 – 2 of 2) sorted by relevance
77 MemoryTypeVector* inp_mtypes, in MemoryTypesForNode() argument93 inp_mtypes->clear(); in MemoryTypesForNode()103 for (const auto& t : inp_dtypes) inp_mtypes->push_back(MTypeFromDType(t)); in MemoryTypesForNode()114 inp_mtypes->resize(GetTotal(inp_names), DEVICE_MEMORY); in MemoryTypesForNode()120 MemoryTypesHelper(inp_names, &host_memory_args, inp_mtypes); in MemoryTypesForNode()127 CHECK_LE(inp_mtypes->size(), inp_dtypes.size()); in MemoryTypesForNode()131 for (int i = 0; i < inp_mtypes->size(); ++i) { in MemoryTypesForNode()133 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()145 if (0 <= i && i < inp_mtypes->size()) { in MemoryTypesForNode()146 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
256 tensorflow::MemoryTypeVector inp_mtypes; in TF_GetSupportedDevices() local259 &inp_mtypes, &out_mtypes); in TF_GetSupportedDevices()261 for (int i = 0; i < inp_mtypes.size(); ++i) { in TF_GetSupportedDevices()262 if (inp_mtypes[i] == tensorflow::HOST_MEMORY) { in TF_GetSupportedDevices()