-
Right now, the program crashes whenever you make a tensor on an XLA device and call a `LazyTensorBarrier()` without setting `wait` to `true`. This can be bypassed by using `LazyTensorBarrier(wait: tru…
-
### Description
`jax.nn.dot_product_attention` does the first dot product with `preferred_element_type=jnp.float32` (see [here](https://github.com/jax-ml/jax/blob/7f655972c47658768b6ecce752fa29c3a…
-
## Fix the Op info test for `nn.functional.leaky_relu .. nn.functional.max_pool1d`
1. Find the lines 233 to 237 of [test_ops.py](test/test_ops.py) and remove
`nn.functional.leaky_relu` .. `nn.func…
qihqi updated
1 month ago
-
## Description
We need to implement a testing framework for the TensorFlow XLA (Accelerated Linear Algebra) compiler model. This framework should comprehensively test the XLA compiler, ensuring it …
-
**Documentation**
[MLIR Bytecode Format](https://mlir.llvm.org/docs/BytecodeFormat/)
**Examples:**
[1043.mlir.zip](https://github.com/lutzroeder/netron/files/14074683/1043.mlir.zip)
[embedding_bm1684…
-
Hello,
We saw the issue that a broadcast tensor from a single-dimension parameter is marked sharded by XLA sharding propagator. This sharded tensor, while doing computation with other tensor which ha…
-
### Description of the bug:
I am using Google Colab Pro+ (with High RAM) to convert Llama 3.2 1B Instruct model to Tflite format (for later use in mediapipe android app). For that
1. I downloaded …
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
Yes
### Source
source
### TensorFlow version
2.18.0-dev20240925
### Custom code
Yes
### OS pla…
x0w3n updated
1 month ago
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
Yes
### Source
source
### TensorFlow version
2.18.0-dev20240925
### Custom code
Yes
### OS platform and distributi…
x0w3n updated
3 weeks ago
-
## Fix the Op info test for `linalg.det`
1. Find the line 113 of [test_ops.py](test/test_ops.py) and remove
`linalg.det` from `skip_list`
2. Run op_info test with `pytest test/test_ops.py`
3. Fix …
qihqi updated
1 month ago