Home
last modified time | relevance | path

Searched refs:nodes_in_memory (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc2467 std::set<std::pair<string, int>> nodes_in_memory; in TEST_F() local
2469 cpu_state.nodes_in_memory.begin(), cpu_state.nodes_in_memory.end(), in TEST_F()
2470 std::inserter(nodes_in_memory, nodes_in_memory.begin()), in TEST_F()
2480 ExpectSetEq(expected, nodes_in_memory); in TEST_F()
Dvirtual_scheduler.cc799 device.nodes_in_memory.insert(std::make_pair(node, port_num)); in MarkCurrNodeExecuted()
835 input_device.nodes_in_memory.erase(std::make_pair(input, port)); in MarkCurrNodeExecuted()
846 device.mem_usage_snapshot_at_peak = device.nodes_in_memory; in MarkCurrNodeExecuted()
Dvirtual_scheduler.h102 nodes_in_memory; member