HomeSort by relevance Sort by last modified time
    Searched defs:live_tensors (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/grappler/costs/
graph_memory.h 40 std::vector<LiveTensor> live_tensors; member in struct:tensorflow::grappler::GraphMemory::MemoryUsage
graph_memory.cc 122 std::unordered_map<string, GraphMemory::LiveTensor*>* live_tensors,
126 auto it = live_tensors->find(name);
127 if (it == live_tensors->end()) {
135 (*live_tensors)[name] = live;
162 std::unordered_map<string, LiveTensor*> live_tensors; local
176 node_stats.node_name(), i, &live_tensors, &device_tensors);
225 input_node, position, &live_tensors,
274 peak_mem_usage.live_tensors.clear();
275 peak_mem_usage.live_tensors.reserve(live_at_peak.size());
277 peak_mem_usage.live_tensors.push_back(*live)
    [all...]

Completed in 52 milliseconds