Searched refs:updates_param (Results 1 – 2 of 2) sorted by relevance
1056 HloInstruction* updates_param = computation_->parameter_instruction(2); in TEST_F() local1064 updates_param, {}, scatter, {})); in TEST_F()
2435 HloInstruction* updates_param = computation_->parameter_instruction(2); in TEST_F() local2443 updates_param, {}, scatter, {})); in TEST_F()