Searched refs:inp_mtypes (Results 1 – 2 of 2) sorted by relevance
84 MemoryTypeVector* inp_mtypes, in MemoryTypesForNode() argument100 inp_mtypes->clear(); in MemoryTypesForNode()123 inp_mtypes->resize(GetTotal(inp_names), DEVICE_MEMORY); in MemoryTypesForNode()129 MemoryTypesHelper(inp_names, &host_memory_args, inp_mtypes); in MemoryTypesForNode()139 inp_mtypes->resize(inp_dtypes.size(), DEVICE_MEMORY); in MemoryTypesForNode()142 CHECK_LE(inp_mtypes->size(), inp_dtypes.size()); in MemoryTypesForNode()146 for (int i = 0; i < inp_mtypes->size(); ++i) { in MemoryTypesForNode()148 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()160 if (0 <= i && i < inp_mtypes->size()) { in MemoryTypesForNode()161 (*inp_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
195 tensorflow::MemoryTypeVector inp_mtypes; in PYBIND11_MODULE() local199 &inp_mtypes, &out_mtypes); in PYBIND11_MODULE()201 for (size_t i = 0; i < inp_mtypes.size(); ++i) { in PYBIND11_MODULE()202 if (inp_mtypes[i] == tensorflow::HOST_MEMORY) { in PYBIND11_MODULE()