StackOverflow Questions for Tag: tensorflow-xla

ldmat
ldmat

Reputation: 1041

Why is JAX jit needed for jax.numpy operations?

Score: 5

Views: 1615

Answers: 1

Read More
ILS
ILS

Reputation: 1380

How to understand the padding rules on cloud TPU?

Score: 0

Views: 291

Answers: 1

Read More
BioGeek
BioGeek

Reputation: 22827

Running Pytorch on Cloud TPU VM on GCP gives INVALID_ARGUMENT: No matching devices found for '/job:localservice/replica:0/task:0/device:TPU_SYSTEM:0'

Score: 0

Views: 391

Answers: 1

Read More
Fred
Fred

Reputation: 666

Tensorflow w/ XLA causing a memory leak

Score: 1

Views: 248

Answers: 0

Read More
StayFoolish
StayFoolish

Reputation: 511

In tensorflow 1.15, what's the difference of using explicit XLA compilation and Auto-clustering?

Score: 1

Views: 820

Answers: 1

Read More
newbie
newbie

Reputation: 463

Tensorflow finding gradients of model output with respect to input on large batch size

Score: 1

Views: 82

Answers: 0

Read More
newbie
newbie

Reputation: 463

Loading a model inside while loop of Tensorflow

Score: 3

Views: 661

Answers: 1

Read More
DavidJ
DavidJ

Reputation: 418

JAX Tridiagonal Jacobians

Score: 1

Views: 276

Answers: 0

Read More
joel
joel

Reputation: 7867

Is XLA's reshape cheap?

Score: 0

Views: 146

Answers: 1

Read More
Jueon Park
Jueon Park

Reputation: 1

Modifying TensorFlow code, especially XLA, yet building TF emits the following error

Score: 0

Views: 175

Answers: 1

Read More
Jueon Park
Jueon Park

Reputation: 1

What does the bracket mean in XLA HLO's tensor shape?

Score: 0

Views: 233

Answers: 1

Read More
Cara Duf
Cara Duf

Reputation: 371

Import error while launching PyTorch Lightning project on Colab TPU

Score: 1

Views: 5637

Answers: 2

Read More
photon1981
photon1981

Reputation: 31

Tensorflow with XLA doesn't fully utilize CPU capacity

Score: 1

Views: 487

Answers: 1

Read More
Raghav Kumar
Raghav Kumar

Reputation: 5

Get C/C++/Latex code from Jaxpr or Jax XLA Computation Object

Score: 1

Views: 295

Answers: 0

Read More
joel
joel

Reputation: 7867

What is XlaBuilder for?

Score: 7

Views: 411

Answers: 1

Read More
joel
joel

Reputation: 7867

How to compose XlaComputation?

Score: 1

Views: 110

Answers: 0

Read More
Aravind Sankaran
Aravind Sankaran

Reputation: 33

Does the TensorFlows XLA compiler optimise matrix chains as a part of its graph optimisation?

Score: 1

Views: 142

Answers: 0

Read More
joel
joel

Reputation: 7867

How to construct an XlaOp?

Score: 1

Views: 147

Answers: 0

Read More
joel
joel

Reputation: 7867

How to return pointer to `GlobalData`?

Score: 0

Views: 68

Answers: 1

Read More
Drphoton
Drphoton

Reputation: 534

JAX/XLA slow compilation using conda

Score: 1

Views: 722

Answers: 0

Read More
PreviousPage 2Next