Avoid JIT compilation errors #121

Job	Run time
static	1m 1s
static	1m 1s
test (test, gpu, tensorflow[and-cuda], self-hosted, gpu)	3m 13s
test (test, gpu, tensorflow[and-cuda], self-hosted, gpu)	5m 16s
test (test, 3.11, latest)	2m 33s
test (test, 3.11, latest)	2m 33s
test (test, tensorflow~=2.8.0, 3.8, oldest)	2m 25s
test (test, tensorflow~=2.8.0, 3.8, oldest)	2m 25s
test (test, tensorflow~=2.9.0, numpy<2, tf-2.9)	2m 19s
test (test, tensorflow~=2.9.0, numpy<2, tf-2.9)	2m 19s
test (test, tensorflow~=2.13.0, tf-2.13)	2m 18s
test (test, tensorflow~=2.13.0, tf-2.13)	2m 18s
test (docs, tensorflow[and-cuda]==2.16.1, 3.9, self-hosted, gpu)	4m 5s
test (docs, tensorflow[and-cuda]==2.16.1, 3.9, self-hosted, gpu)	4m 5s
test (examples, tensorflow[and-cuda]==2.16.1, self-hosted, gpu)	3m 45s
test (examples, tensorflow[and-cuda]==2.16.1, self-hosted, gpu)	3m 45s
coverage	21s
coverage	20s
deploy	0s
deploy	0s
	46m 2s

Provide feedback