Searched refs:peak_buffers (Results 1 – 1 of 1) sorted by relevance
1848 const std::vector<const HloValue*>& peak_buffers = in TEST_F() local1850 ASSERT_EQ(peak_buffers.size(), 1); in TEST_F()1851 EXPECT_EQ(peak_buffers[0]->instruction(), sub); in TEST_F()1895 const std::vector<const HloValue*>& peak_buffers = in TEST_F() local1899 ASSERT_EQ(peak_buffers.size(), 3); in TEST_F()1901 for (const HloValue* logical_buffer : peak_buffers) { in TEST_F()