| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 213917946
|
|
|
|
| |
PiperOrigin-RevId: 213917881
|
|
|
|
| |
PiperOrigin-RevId: 213915666
|
|
|
|
| |
PiperOrigin-RevId: 213913013
|
|\
| |
| |
| | |
PiperOrigin-RevId: 213912651
|
| |
| |
| |
| | |
PiperOrigin-RevId: 213912507
|
|\ \
| | |
| | |
| | | |
PiperOrigin-RevId: 213908983
|
| | |
| | |
| | |
| | |
| | |
| | | |
refactoring the API for exposing tunable parameters, and removing `model::Node` from the public API.
PiperOrigin-RevId: 213907565
|
|\ \ \
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213906379
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
depthwise convolution instead of a full convolution now that it exists in XLA.
PiperOrigin-RevId: 213896333
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213896057
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
This CL splits the functionality in XlaLaunch into two separate operations:
- XlaCompile, responsible for compiling a TF function into a LocalExecutable
- XlaRun, responsible for executing a LocalExecutable created by XlaCompile
This CL is a stepping stone towards implementing lazy compilation for TF/XLA.
The XlaCompile op is spec'ed to return a boolean indicating whether the
compilation was successful. Right now that boolean is always set to true by
XlaCompile and its value is otherwise ignored, but in the future it will be used
to indicate whether the TF function was compiled or not, and thus whether we
should execute XlaRun or just directly call the TF function.
XlaLaunch still exists, and will be created by create_xla_launch_op.cc. In the
future we may consider removing it altogether. build_xla_launch_ops.cc, now
renamed to build_xla_ops.cc, creates a XlaCompile/XlaRun pair instead of
XlaLaunch.
This CL is organized as follows:
- jit/ops/xla_ops.cc gets two new XLA-specific operations, XlaCompile and
XlaRun, described above. XlaRun redundantly takes the must-be-constant
inputs to the TensorFlow cluster to keep the implementation simple (simple in
the sense of similar to XlaLaunch), but I will remove this in a subsequent
cleanup CL.
- jit/kernels/xla_ops.cc implements XlaCompile and XlaRun in a fairly
straightforward manner. XlaCompile compiles the TF function, puts it in a
process-global storage, XlaExecutableClosureStore, and produces a int64 key.
XlaRun uses the key to read out the LocalExecutable and execute it. I'm not
sure if XlaExecutableClosureStore should be a resource like
XlaCompilationCache; I did not immediately see any reason to make it so.
- There are changes to the various _device files to register XlaCompile and
XlaRun for the XLA_* devices.
- Finally, I had to fix some tests that were expecting XlaLaunch in the
execution timeline.
PiperOrigin-RevId: 213895405
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
environment configurations to a separate test target, and disables running them on Windows.
PiperOrigin-RevId: 213895372
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213890403
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213886813
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213885561
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
to be of the form 'scope/<arbitrary_text>'. Relax restriction to handle empty scopes. Enable this change to work for both fused and unfused batch norm layers
PiperOrigin-RevId: 213883621
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
The reshape we created change the element type unintentionally.
PiperOrigin-RevId: 213883142
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213882538
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
not found in the namespace in Python 3.
PiperOrigin-RevId: 213879813
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
Minor cleanup by moving the helper function ShapesEqual to GraphProperties and adding unit tests for it.
PiperOrigin-RevId: 213876779
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213875284
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213873471
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
1. In ParallelMapIterator, do not call `cond_var_.notify_all()` without holding
the associated mutex. In some cases, the iterator may have been deleted
between releasing the lock and notifying the condition variable, which
leads to a use-after-free. This change applies this style to all use of
condition variables in tensorflow/core/kernels/data/.
2. In CapturedFunction::RunAsync(), do not use `shared_ptr` to manage
the lifetime of objects that (potentially) borrow from runtime
objects. The present code runs the destructor after the `done()`
callback is called, but the `done()` callback may be the last
action in a session, and thus trigger destruction of those borrowed
objects. In that case, the `shared_ptr` destructor may use the
borrowed objects after they are freed.
PiperOrigin-RevId: 213872829
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
detection models.
As part of this CL, we use the Keras mobilenet_v2 application's keyword argument layer injection API to allow the generated network to support the object detection hyperparameters.
PiperOrigin-RevId: 213872175
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213872127
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213867606
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213866466
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 213863392
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
https://github.com/tensorflow/community/pull/13
PiperOrigin-RevId: 213862844
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
problematic to use in eager because of the circular references it creates.
PiperOrigin-RevId: 213862402
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
refactoring it, adding several new fields and an EmbeddingOutputLayout message to provide experimental support for controlling the embedding output.
PiperOrigin-RevId: 213849572
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
number of filtered elements to monitoring counter.
PiperOrigin-RevId: 213846793
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
With the exception of StrCat all of these are using absl already, this change
just removes one layer of indirection.
PiperOrigin-RevId: 213846036
|
|\ \ \ \
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 213844688
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
optimization
PiperOrigin-RevId: 213840320
|
|\ \ \ \ \
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 213836802
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 213829360
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 213801006
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
These have the same behavior as unquantized types so we can just pass them
through to XLA (which converts them to unquantized types). They're supposed to
be used with special ops, none of which are currently implemented by XLA.
Casting (without quantization) and basic math works fine though.
These do not have a corresponding numpy type, so only tests using TF types will
see them.
PiperOrigin-RevId: 213781650
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 213773990
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 213771631
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 213770000
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 213764810
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
The only tensorflow op that uses XlaSort is nn.top_k, so we add a test case
using nn.top_k.
PiperOrigin-RevId: 213763591
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
This parameter has been added to HLO to support depthwise convolution.
PiperOrigin-RevId: 213761790
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 213753728
|
| |_|_|_|/
|/| | | | |
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 213749129
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 213737482
|