| Commit message (Collapse) | Author | Age |
|
|
|
|
|
|
|
| |
self.test_session() has been deprecated in 9962eb5e84b15e309410071b06c2ed2d6148ed44 as its name confuses readers of the test. Moving to cached_session() instead which is more explicit about:
* the fact that the session may be reused.
* the session is not closed even when doing a "with self.test_session()" statement.
PiperOrigin-RevId: 209837298
|
|
|
|
| |
PiperOrigin-RevId: 202572322
|
|
|
|
|
|
|
|
|
|
| |
relevant elementwise unary and binary op lowering for CPU and GPU.
We use a named LLVM struct "complex64", laid out the same as std::complex<float>. This named struct is accessed via the llvm::Module, which required changes to accessors of PrimitiveTypeToIrType & friends.
Ops that require atan2 (in particular, angle and log) are only supported on GPU at this point. LLVM lacks a CPU intrinsic for atan or atan2, whereas libdevice provides this for GPU.
PiperOrigin-RevId: 173676849
|
|
|
|
| |
PiperOrigin-RevId: 171025171
|
|
|
|
| |
PiperOrigin-RevId: 170754745
|
|
|
|
| |
Change: 152200430
|
|
|
|
|
| |
Fix crash in normal OneHot kernel for depth < 0.
Change: 148881102
|
|
|
|
| |
Change: 144153795
|
|
XLA is a compiler-based linear algebra execution engine that targets CPUs, GPUs and custom accelerators.
XLA is still experimental; we are releasing it early to get the community involved.
Change: 143990941
|