Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dvirtual_scheduler_test.cc2541 ASSERT_EQ(cpu_state.temporary_memory_usage_trace.size(), 10); in TEST_F()
2543 cpu_state.temporary_memory_usage_trace.at(4); in TEST_F()
2547 cpu_state.temporary_memory_usage_trace.at(8); in TEST_F()
2551 cpu_state.temporary_memory_usage_trace.at(9); in TEST_F()
H A Dvirtual_scheduler.h136 std::vector<std::pair<std::string, int64_t>> temporary_memory_usage_trace; member
H A Dvirtual_scheduler.cc1015 device.temporary_memory_usage_trace.push_back( in MarkNodeExecuted()