Searched refs:test_constraint (Results 1 – 2 of 2) sorted by relevance
102 def test_constraint(constraint_fn, result, value, is_cuda): function
1373 def test_constraint(self): member in TPUStrategyDistributionTest