Home
last modified time | relevance | path

Searched refs:out_mtypes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dmemory_types.cc84 MemoryTypeVector* out_mtypes) { in MemoryTypesForNode() argument
100 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 …]
/external/tensorflow/tensorflow/python/grappler/
Dcluster_wrapper.cc195 tensorflow::MemoryTypeVector out_mtypes; in PYBIND11_MODULE() local
198 &inp_mtypes, &out_mtypes); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc387 tensorflow::MemoryTypeVector out_mtypes; in IsHostMemory() local
389 node, &in_mtypes, &out_mtypes); in IsHostMemory()
391 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()
Dgeneric_layout_optimizer_transposer.cc132 tensorflow::MemoryTypeVector out_mtypes; in IsHostMemory() local
134 node, &in_mtypes, &out_mtypes); in IsHostMemory()
136 if (out_mtypes[output_port] == HOST_MEMORY) { in IsHostMemory()