Searched refs:out_mtypes (Results 1 – 4 of 4) sorted by relevance
84 MemoryTypeVector* out_mtypes) { in MemoryTypesForNode() argument100 out_mtypes->clear(); in MemoryTypesForNode()123 out_mtypes->push_back(MTypeFromDTypeIntsOnDevice(t)); in MemoryTypesForNode()126 for (const auto& t : out_dtypes) out_mtypes->push_back(MTypeFromDType(t)); in MemoryTypesForNode()138 out_mtypes->resize(GetTotal(out_names), DEVICE_MEMORY); in MemoryTypesForNode()144 MemoryTypesHelper(out_names, &host_memory_args, out_mtypes); in MemoryTypesForNode()151 CHECK_LE(out_mtypes->size(), out_dtypes.size()); in MemoryTypesForNode()159 for (int i = 0; i < out_mtypes->size(); ++i) { in MemoryTypesForNode()161 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()175 if (0 <= i && i < out_mtypes->size()) { in MemoryTypesForNode()[all …]
195 tensorflow::MemoryTypeVector out_mtypes; in PYBIND11_MODULE() local198 &inp_mtypes, &out_mtypes); in PYBIND11_MODULE()
387 tensorflow::MemoryTypeVector out_mtypes; in IsHostMemory() local389 node, &in_mtypes, &out_mtypes); in IsHostMemory()391 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()
132 tensorflow::MemoryTypeVector out_mtypes; in IsHostMemory() local134 node, &in_mtypes, &out_mtypes); in IsHostMemory()136 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()