Searched refs:xla_allow_fallback (Results 1 – 5 of 5) sorted by relevance
275 test_util.xla_allow_fallback(283 test_util.xla_allow_fallback(
578 @test_util.xla_allow_fallback("XLA doesn't support explicit padding")591 @test_util.xla_allow_fallback("XLA doesn't support explicit padding")605 @test_util.xla_allow_fallback("XLA doesn't support explicit padding")619 @test_util.xla_allow_fallback("XLA doesn't support explicit padding")634 @test_util.xla_allow_fallback("XLA doesn't support explicit padding")651 @test_util.xla_allow_fallback("XLA doesn't support explicit padding")666 @test_util.xla_allow_fallback("XLA doesn't support explicit padding")
64 @test_util.xla_allow_fallback("b/128495870")
5180 @test_util.xla_allow_fallback(5203 @test_util.xla_allow_fallback(5226 @test_util.xla_allow_fallback(5316 @test_util.xla_allow_fallback(5440 @test_util.xla_allow_fallback(5468 @test_util.xla_allow_fallback(
2239 def xla_allow_fallback(description): # pylint: disable=unused-argument function