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.cc2638 std::set<std::pair<string, int>> nodes_in_memory; in TEST_F() local
2640 cpu_state.nodes_in_memory.begin(), cpu_state.nodes_in_memory.end(), in TEST_F()
2641 std::inserter(nodes_in_memory, nodes_in_memory.begin()), in TEST_F()
2651 ExpectSetEq(expected, nodes_in_memory); in TEST_F()
Dvirtual_scheduler.cc940 device.nodes_in_memory.insert(std::make_pair(node, port_num)); in MarkNodeExecuted()
973 device.mem_usage_snapshot_at_peak = device.nodes_in_memory; in MarkNodeExecuted()
1001 input_device.nodes_in_memory.erase(std::make_pair(input, port)); in MarkNodeExecuted()
Dvirtual_scheduler.h117 nodes_in_memory; member