Searched refs:host_memory_args (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.cc | 44 std::vector<string>* host_memory_args, in MemoryTypesHelper() argument
48 for (size_t i = 0; i < host_memory_args->size(); ++i) { in MemoryTypesHelper()
49 auto iter = name_map.find((*host_memory_args)[i]); in MemoryTypesHelper()
56 if (i > keep) (*host_memory_args)[keep] = (*host_memory_args)[i]; in MemoryTypesHelper()
60 host_memory_args->resize(keep); in MemoryTypesHelper()
119 std::vector<string> host_memory_args(from_proto.begin(), from_proto.end()); in MemoryTypesForNode() local
120 MemoryTypesHelper(inp_names, &host_memory_args, inp_mtypes); in MemoryTypesForNode()
121 MemoryTypesHelper(out_names, &host_memory_args, out_mtypes); in MemoryTypesForNode()
122 if (!host_memory_args.empty()) { in MemoryTypesForNode()
124 "HostMemory args '", str_util::Join(host_memory_args, "', '"), in MemoryTypesForNode()
|