Home
last modified time | relevance | path

Searched refs:set_schedule (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc308 TF_CHECK_OK(module->set_schedule(schedule)); in CreateEvictAndPrefetchModule()
414 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
443 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
499 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
630 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
718 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
809 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
875 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
911 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
949 TF_CHECK_OK(module->set_schedule(schedule)); in TEST_P()
[all …]
Dhlo_module.cc52 Status HloModule::set_schedule(HloSchedule schedule) { in set_schedule() function in xla::HloModule
422 TF_RETURN_IF_ERROR(module->set_schedule(std::move(schedule))); in CreateFromProto()
729 TF_CHECK_OK(module->set_schedule(std::move(clone_schedule))); in Clone()
Dhlo_module.h304 Status set_schedule(HloSchedule schedule);
Dhlo_memory_scheduler.cc736 TF_RETURN_IF_ERROR(module->set_schedule(std::move(schedule))); in Run()
755 TF_RETURN_IF_ERROR(module->set_schedule(std::move(schedule))); in Run()
Dhlo_memory_scheduler_test.cc150 TF_ASSERT_OK(module->set_schedule(schedule)); in TEST_F()
Dhlo_rematerialization.cc2080 TF_RETURN_IF_ERROR(module->set_schedule(std::move(saved_schedule))); in Run()
Dhlo_parser.cc815 TF_CHECK_OK(module->set_schedule(ScheduleFromInstructionOrder(module))); in ParseHloModule()
5239 TF_CHECK_OK(module->set_schedule(ScheduleFromInstructionOrder(module))); in ParseSingleInstruction()
/external/llvm-project/polly/lib/External/isl/include/isl/
Dcpp-checked.h2674 inline isl::checked::union_access_info set_schedule(isl::checked::schedule schedule) const;
10948 isl::checked::union_access_info union_access_info::set_schedule(isl::checked::schedule schedule) co… in set_schedule() function
Disl-noexceptions.h2997 inline union_access_info set_schedule(schedule schedule) const;
16204 union_access_info union_access_info::set_schedule(schedule schedule) const in set_schedule() function
Dcpp.h2739 inline isl::union_access_info set_schedule(isl::schedule schedule) const;
15411 isl::union_access_info union_access_info::set_schedule(isl::schedule schedule) const in set_schedule() function
/external/llvm-project/polly/lib/External/isl/interface/
Disl.py13502 def set_schedule(arg0, arg1): member in union_access_info