Searched refs:inp_mtypes (Results 1 – 2 of 2) sorted by relevance
83 MemoryTypeVector* inp_mtypes, in MemoryTypesForNode() argument99 inp_mtypes->clear(); in MemoryTypesForNode()121 inp_mtypes->push_back(MTypeFromDTypeIntsOnDevice(t)); in MemoryTypesForNode()125 for (const auto& t : inp_dtypes) inp_mtypes->push_back(MTypeFromDType(t)); in MemoryTypesForNode()137 inp_mtypes->resize(GetTotal(inp_names), DEVICE_MEMORY); in MemoryTypesForNode()143 MemoryTypesHelper(inp_names, &host_memory_args, inp_mtypes); in MemoryTypesForNode()150 CHECK_LE(inp_mtypes->size(), inp_dtypes.size()); in MemoryTypesForNode()154 for (int i = 0; i < inp_mtypes->size(); ++i) { in MemoryTypesForNode()156 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()168 if (0 <= i && i < inp_mtypes->size()) { in MemoryTypesForNode()[all …]
194 tensorflow::MemoryTypeVector inp_mtypes; in PYBIND11_MODULE() local198 &inp_mtypes, &out_mtypes); in PYBIND11_MODULE()200 for (int i = 0; i < inp_mtypes.size(); ++i) { in PYBIND11_MODULE()201 if (inp_mtypes[i] == tensorflow::HOST_MEMORY) { in PYBIND11_MODULE()