Searched refs:host_memory_args (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.cc | 46 std::vector<string>* host_memory_args, in MemoryTypesHelper() argument
50 for (size_t i = 0; i < host_memory_args->size(); ++i) { in MemoryTypesHelper()
51 auto iter = name_map.find((*host_memory_args)[i]); in MemoryTypesHelper()
58 if (i > keep) (*host_memory_args)[keep] = (*host_memory_args)[i]; in MemoryTypesHelper()
62 host_memory_args->resize(keep); in MemoryTypesHelper()
128 std::vector<string> host_memory_args(from_proto.begin(), from_proto.end()); in MemoryTypesForNode() local
129 MemoryTypesHelper(inp_names, &host_memory_args, inp_mtypes); in MemoryTypesForNode()
130 MemoryTypesHelper(out_names, &host_memory_args, out_mtypes); in MemoryTypesForNode()
131 if (!host_memory_args.empty()) { in MemoryTypesForNode()
133 "HostMemory args '", absl::StrJoin(host_memory_args, "', '"), in MemoryTypesForNode()
|