Home
last modified time | relevance | path

Searched refs:is_computation_scheduled (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dinstruction_hoister.cc33 CHECK(schedule.is_computation_scheduled(computation)); in HoistParameters()
87 CHECK(schedule.is_computation_scheduled(computation)); in HoistConstantOperations()
H A Dmemory_space_assignment_utils.cc95 CHECK(schedule.is_computation_scheduled(computation)); in HoistConstantOperations()
H A Dhlo_schedule.h145 bool is_computation_scheduled(const HloComputation* computation) const { in is_computation_scheduled() function
H A Dasync_collective_creator.cc60 module->schedule().is_computation_scheduled(computation); in Run()
H A Dhlo_schedule_test.cc416 ASSERT_FALSE(schedule.is_computation_scheduled( in TEST_F()
H A Dhlo_module.cc290 if (has_schedule() && schedule().is_computation_scheduled(computation)) { in ToCord()
883 if (schedule().is_computation_scheduled(computation)) { in Clone()
H A Dhlo_ordering.cc533 return schedule_.is_computation_scheduled(&computation) in SequentialOrder()
H A Dhlo_module_test.cc371 ASSERT_TRUE(module_copy->schedule().is_computation_scheduled( in TEST_F()
H A Dmemory_space_assignment.cc4103 TF_RET_CHECK(schedule.is_computation_scheduled(computation)); in FixSchedule()