Searched refs:mom_out (Results 1 – 1 of 1) sorted by relevance
346 auto mom_out = builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F() local353 shape, HloOpcode::kSubtract, var, mom_out)); in TEST_F()356 all_outputs.push_back(mom_out); in TEST_F()