Home
last modified time | relevance | path

Searched refs:GetComputationWithName (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_constant_sinking_test.cc64 auto* while_body = module->GetComputationWithName("body"); in TEST_F()
104 auto* while_body = module->GetComputationWithName("body"); in TEST_F()
145 auto* while_body = module->GetComputationWithName("body"); in TEST_F()
193 auto* while_body = module->GetComputationWithName("body"); in TEST_F()
235 auto* while_body = module->GetComputationWithName("body"); in TEST_F()
281 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F()
321 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F()
364 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F()
414 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F()
Dwhile_util_test.cc157 HloComputation* while_body = module->GetComputationWithName("body"); in TEST()
198 HloComputation* main = module->GetComputationWithName("main"); in TEST()
Dhlo_module.h165 HloComputation* GetComputationWithName(absl::string_view name);
Dwhile_loop_invariant_code_motion_test.cc501 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F()
614 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F()
Dtranspose_folding_test.cc211 const HloComputation* callee = module->GetComputationWithName("callee"); in TEST_F()
Dhlo_module.cc611 HloComputation* HloModule::GetComputationWithName(absl::string_view name) { in GetComputationWithName() function in xla::HloModule
Dbuffer_assignment_test.cc2352 HloComputation* callee = m->GetComputationWithName("Callee"); in TEST_F()