tensorflow-xla Questions
3
I got this when using keras with Tensorflow backend:
tensorflow.python.framework.errors_impl.InvalidArgumentError: device CUDA:0 not supported by XLA service
while setting up XLA_GPU_JIT device...
Overtake asked 13/8, 2019 at 18:33
1
Solved
From the JAX docs:
import jax
import jax.numpy as jnp
def selu(x, alpha=1.67, lambda_=1.05):
return lambda_ * jnp.where(x > 0, x, alpha * jnp.exp(x) - alpha)
x = jnp.arange(1000000)
selu(x)
...
Narine asked 16/1, 2023 at 18:43
1
Solved
What's the XLA class XlaBuilder for? The docs describe its interface but don't provide a motivation.
The presentation in the docs, and indeed the comment above XlaBuilder in the source code
// A co...
Folderol asked 13/12, 2021 at 19:11
1
I'm interested in using XLA for the training with the custom Device (FPGA, ...).
However, I learned that XLA is now in experimental status from developer's tutorial.
https://www.tensorflow.org/per...
Involucre asked 9/12, 2017 at 7:53
0
Can anyone give me any hints why XLA-JIT has better performance on CPU backend?
I tried TensorFlow without and with XLA-JIT (manual mode) on mnist benchmark on a single CPU. Using XLA-JIT achieves ...
Aunt asked 24/11, 2017 at 4:29
1
Solved
Here's an example to clarify what I mean:
First session.run():
First run of a TensorFlow session
Later session.run():
Later runs of a TensorFlow session
I understand TensorFlow is doing some init...
Gaiseric asked 12/7, 2017 at 16:49
1
© 2022 - 2024 — McMap. All rights reserved.