Searched refs:out_mtypes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types.cc | 85 MemoryTypeVector* out_mtypes) { in MemoryTypesForNode() argument
101 out_mtypes->clear(); in MemoryTypesForNode()
124 out_mtypes->resize(GetTotal(out_names), DEVICE_MEMORY); in MemoryTypesForNode()
130 MemoryTypesHelper(out_names, &host_memory_args, out_mtypes); in MemoryTypesForNode()
140 out_mtypes->resize(out_dtypes.size(), DEVICE_MEMORY); in MemoryTypesForNode()
143 CHECK_LE(out_mtypes->size(), out_dtypes.size()); in MemoryTypesForNode()
151 for (int i = 0; i < out_mtypes->size(); ++i) { in MemoryTypesForNode()
153 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
168 if (0 <= i && i < out_mtypes->size()) { in MemoryTypesForNode()
169 (*out_mtypes)[i] = HOST_MEMORY; in MemoryTypesForNode()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 196 tensorflow::MemoryTypeVector out_mtypes; in PYBIND11_MODULE() local
199 &inp_mtypes, &out_mtypes); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 388 tensorflow::MemoryTypeVector out_mtypes; in IsHostMemory() local
390 node, &in_mtypes, &out_mtypes); in IsHostMemory()
392 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()
|
| D | generic_layout_optimizer_transposer.cc | 146 tensorflow::MemoryTypeVector out_mtypes; in IsHostMemory() local
148 node, &in_mtypes, &out_mtypes); in IsHostMemory()
150 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()
|