Searched refs:scalar_param (Results 1 – 2 of 2) sorted by relevance
277 auto* scalar_param = builder.AddInstruction( in TEST_F() local281 HloInstruction::CreateTuple({scalar_param, scalar_param, token})); in TEST_F()330 auto* scalar_param = builder.AddInstruction( in TEST_F() local334 HloInstruction::CreateTuple({scalar_param, scalar_param, token})); in TEST_F()
3508 HloInstruction* scalar_param = builder.AddInstruction( in TEST_F() local3513 HloInstruction::CreateBroadcast(broadcast_shape, scalar_param, {})); in TEST_F()3533 GmockMatch(m::Broadcast(m::Op().Is(scalar_param)) in TEST_F()