Searched refs:reshape_pred (Results 1 – 1 of 1) sorted by relevance
432 auto reshape_pred = in TEST_F() local435 root_shape, HloOpcode::kSelect, reshape_pred, reshape0, reshape1)); in TEST_F()462 auto reshape_pred = in TEST_F() local465 root_shape, HloOpcode::kSelect, reshape_pred, param0, param1)); in TEST_F()