Searched refs:hostmem_attr (Results 1 – 1 of 1) sorted by relevance
165 std::vector<int32> hostmem_attr; in MemoryTypesForNode() local166 if (TryGetNodeAttr(ndef, "_input_hostmem", &hostmem_attr)) { in MemoryTypesForNode()167 for (int32 i : hostmem_attr) { in MemoryTypesForNode()173 if (TryGetNodeAttr(ndef, "_output_hostmem", &hostmem_attr)) { in MemoryTypesForNode()174 for (int32 i : hostmem_attr) { in MemoryTypesForNode()