Searched refs:GetComputationWithName (Results 1 – 11 of 11) sorted by relevance
63 auto* while_body = module->GetComputationWithName("body"); in TEST_F()110 auto* while_body = module->GetComputationWithName("body"); in TEST_F()150 auto* while_body = module->GetComputationWithName("body"); in TEST_F()191 auto* while_body = module->GetComputationWithName("body"); in TEST_F()239 auto* while_body = module->GetComputationWithName("body"); in TEST_F()281 auto* while_body = module->GetComputationWithName("body"); in TEST_F()327 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F()367 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F()410 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F()460 auto* while_condition = module->GetComputationWithName("condition"); in TEST_F()
78 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F()100 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F()184 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F()234 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F()
162 HloComputation* while_body = module->GetComputationWithName("body"); in TEST_F()203 HloComputation* main = module->GetComputationWithName("main"); in TEST_F()
204 cloned_module->GetComputationWithName("add_s32.clone"); in TEST_F()242 cloned_module->GetComputationWithName("add_s32_0.clone"); in TEST_F()244 cloned_module->GetComputationWithName("add_s32_1.clone"); in TEST_F()272 cloned_module->GetComputationWithName("fused_computation.clone"); in TEST_F()
501 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F()614 HloComputation* while_body = m->GetComputationWithName("wide.body"); in TEST_F()
2045 module->GetComputationWithName("entry")->GetInstructionWithName("copy"); in TEST_P()2047 auto neg1 = module->GetComputationWithName("true_computation") in TEST_P()2052 auto neg2 = module->GetComputationWithName("false_computation") in TEST_P()2108 module->GetComputationWithName("entry")->GetInstructionWithName( in TEST_P()2112 module->GetComputationWithName("entry")->GetInstructionWithName( in TEST_P()2115 auto add = module->GetComputationWithName("true_computation") in TEST_P()2172 module->GetComputationWithName("entry")->GetInstructionWithName( in TEST_P()2175 auto add0 = module->GetComputationWithName("true_computation") in TEST_P()2181 module->GetComputationWithName("entry")->GetInstructionWithName("add1"); in TEST_P()2241 module->GetComputationWithName("entry")->GetInstructionWithName( in TEST_P()[all …]
239 HloComputation* GetComputationWithName(absl::string_view name);
127 module->GetComputationWithName("reduction"); in TEST_F()
218 const HloComputation* callee = module->GetComputationWithName("callee"); in TEST_F()
963 HloComputation* HloModule::GetComputationWithName(absl::string_view name) { in GetComputationWithName() function in xla::HloModule
2577 HloComputation* callee = m->GetComputationWithName("Callee"); in TEST_F()