Avoid JIT compilation errors #119

Job	Run time
static	1m 4s
static	1m 4s
test (test, gpu, tensorflow[and-cuda], self-hosted, gpu)	2m 25s
test (test, gpu, tensorflow[and-cuda], self-hosted, gpu)	5m 29s
test (test, 3.11, latest)	3m 38s
test (test, 3.11, latest)	3m 38s
test (test, tensorflow~=2.8.0, 3.8, oldest)	2m 39s
test (test, tensorflow~=2.8.0, 3.8, oldest)	2m 39s
test (test, tensorflow~=2.9.0, numpy<2, tf-2.9)	2m 31s
test (test, tensorflow~=2.9.0, numpy<2, tf-2.9)	2m 31s
test (test, tensorflow~=2.13.0, tf-2.13)	4m 2s
test (test, tensorflow~=2.13.0, tf-2.13)	4m 2s
test (docs, tensorflow[and-cuda]==2.16.1, 3.9, self-hosted, gpu)	2m 42s
test (docs, tensorflow[and-cuda]==2.16.1, 3.9, self-hosted, gpu)	2m 42s
test (examples, tensorflow[and-cuda]==2.16.1, self-hosted, gpu)	2m 27s
test (examples, tensorflow[and-cuda]==2.16.1, self-hosted, gpu)	2m 27s
coverage	25s
coverage	24s
deploy	0s
deploy	0s
	46m 49s

Provide feedback