Home
last modified time | relevance | path

Searched refs:cross_program_prefetches (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc4853 auto cross_program_prefetches = module->CrossProgramPrefetches(); in TEST_P() local
4854 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() local
4900 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 …]
Dhlo_module.cc425 for (auto prefetch : proto.cross_program_prefetches()) { in CreateFromProto()
Dhlo.proto402 repeated CrossProgramPrefetch cross_program_prefetches = 10; field
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc472 for (auto& prefetch : hlo_metadata.hlo_module().cross_program_prefetches()) { in TPUExecute()