Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dmemory_types.cc142 std::vector<int32> hostmem_attr; in MemoryTypesForNode() local
143 if (GetNodeAttr(ndef, "_input_hostmem", &hostmem_attr).ok()) { in MemoryTypesForNode()
144 for (int32 i : hostmem_attr) { in MemoryTypesForNode()
150 if (GetNodeAttr(ndef, "_output_hostmem", &hostmem_attr).ok()) { in MemoryTypesForNode()
151 for (int32 i : hostmem_attr) { in MemoryTypesForNode()