| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 202572322
|
|
|
|
|
|
| |
ResourceScatter{Update,Add,Sub,Mul,Div,Min,Max,NdAdd,NdUpdate}.
PiperOrigin-RevId: 201723512
|
|
|
|
|
|
| |
This allows us to avoid an XLA compilation and tensor copies when assigning to a variable placed on an XLA device.
PiperOrigin-RevId: 198127062
|
|
|
|
|
|
| |
Skip individual tests that are currently failing with bfloat16.
PiperOrigin-RevId: 191296618
|
|
|
|
| |
PiperOrigin-RevId: 178084701
|
|
|
|
|
|
|
|
|
|
| |
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: 167902092
|
|
|
|
| |
PiperOrigin-RevId: 161730154
|
|
|
|
| |
PiperOrigin-RevId: 156341053
|
|
|
|
|
|
| |
and one or more variable outputs.
PiperOrigin-RevId: 156193188
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
* enable compilation of VarIsInitializedOp.
* fix deprecated variable initializer in variable_ops_test.py
* simplify variable logic in XlaContext, move intelligence into XlaOpKernelContext.
* add resource variable support in the contrib layers library.
Cleanups and refactorings:
* merge XlaCompiler::CompileSubComputation with XlaCompiler::CompileFunction.
* pass XlaCompiler arguments consistently via XlaCompiler::Options.
* split the two roles of XlaCompiler::CompilationResult::input_shapes into input_mapping and xla_input_shapes.
* initialize the numpy and Python seeds to a constant for XLA test cases.
Change: 148683645
|
|
Change: 148176223
|