Searched refs:cross_program_prefetches (Results 1 – 4 of 4) sorted by relevance
4853 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P() local4854 EXPECT_EQ(cross_program_prefetches.size(), 1); in TEST_P()4855 if (!cross_program_prefetches.empty()) { in TEST_P()4856 EXPECT_EQ(cross_program_prefetches[0].first, 0); in TEST_P()4857 EXPECT_EQ(cross_program_prefetches[0].second, ShapeIndex({1})); in TEST_P()4899 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P() local4900 EXPECT_EQ(cross_program_prefetches.size(), 1); in TEST_P()4901 if (!cross_program_prefetches.empty()) { in TEST_P()4902 EXPECT_EQ(cross_program_prefetches[0].first, 0); in TEST_P()4903 EXPECT_EQ(cross_program_prefetches[0].second, ShapeIndex({1})); in TEST_P()[all …]
425 for (auto prefetch : proto.cross_program_prefetches()) { in CreateFromProto()
402 repeated CrossProgramPrefetch cross_program_prefetches = 10; field
472 for (auto& prefetch : hlo_metadata.hlo_module().cross_program_prefetches()) { in TPUExecute()