-
## Fix the Op info test for `nn.functional.celu .. nn.functional.glu`
1. Find the lines 418 to 422 of [test_ops.py](test/test_ops.py) and remove
`nn.functional.celu` .. `nn.functional.glu` from `s…
-
This is mostly a reminder for myself, but also documentation if someone else ends up doing it.
After reading through the python code (see [here](https://github.com/tensorflow/tensorflow/blob/4070c4…
rnett updated
4 months ago
-
## 📚 Documentation
This isn't necessarily an issue with the documentation, but an inconsistency between the documentation and the simplest [Pytorch XLA example](https://github.com/pytorch/xla/blob/…
-
PR #7318 introduced OpenXLA CUDA fallback. However, that's an opt-in feature that relies on the environment variable: `XLA_FALLBACK_CUDA`. Even though that works fine, as [this comment suggests](https…
-
Hello, I am using horovod docker for training deep learning models:
ENV:nvcr.io/nvidia/tensorflow:20.06-tf1-py3
WARNING:tensorflow:From /dockerdata/trainer/apd_model.py:112: The name tf.summary.…
-
For some reason, `torch.argsort` is crashing when the tensor is on a Neuron device. For example, the code snippet bellow works as expected (on CPU):
```
>>> import torch
>>> import torch_neuronx
>…
-
## Fix the Op info test for `native_dropout_backward .. nn.functional.binary_cross_entropy_with_logits`
1. Find the lines 413 to 417 of [test_ops.py](test/test_ops.py) and remove
`native_dropout_b…
-
## ❓ Questions and Help
Currently `F.interpolate(mode='tilinear)'` will be dispatched to `aten::upsample_trilinear3d` which we don't have c++ lowering. There is a python decomp for this op in https:/…
-
The current version is not detecting the TPU environment on Kaggle.
example:
https://github.com/huggingface/notebooks/blob/main/examples/accelerate_examples/simple_cv_example.ipynb
with run: …
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
Yes
### Source
binary
### TensorFlow version
v2.14.0-4248-g3448956e87e 2.14.0.600
### Custom code
Yes
### OS platf…