| Commit message (Collapse) | Author | Age |
|
|
|
|
|
|
|
|
|
| |
Instead of symlinking the install dir, copy the two files we need.
Symlinking a system dir like /usr is generally problematic as it can quickly
lead to miscompiles for unrelated reasons. Furthermore, bazel will consider
it an error if /usr is linked in and contains a recursive symlink in
/usr/bin/X11 -> .
PiperOrigin-RevId: 211842260
|
|
|
|
| |
PiperOrigin-RevId: 211842211
|
|
|
|
| |
PiperOrigin-RevId: 211840928
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
Moved _is_free_function to parser.is_free_function
Merged the `is_class` and `is_module` properties into `is_fragment`, since this is the only thing they were being used for.
With the additions to `pretty_docs.py`, all documented objects either have a page to them self, or a `#id` fragment on their parents page, the `is_fragment` property indicates which.
In all uses of `documentation_path`, except the "reference_to_url" it's safe to assume that `is_fragment` is `False` (this is the current correct behavior).
fixes #20913
PiperOrigin-RevId: 211838909
|
|
|
|
|
|
|
|
|
| |
the regex pattern are fixed.
This allows the Op to perform the expensive regex compilation once upon creation instead of with each call to compute.
RELNOTES: Performance improvements for regex full match operations.
PiperOrigin-RevId: 211835278
|
|
|
|
|
|
|
|
| |
* Allow sparse labels to work with Datasets.
* Allow sample_weights to be passed as the third output of a Dataset (like how
generator input is treated).
PiperOrigin-RevId: 211834259
|
|
|
|
| |
PiperOrigin-RevId: 211833891
|
|
|
|
|
|
| |
lightweight way
PiperOrigin-RevId: 211833556
|
|
|
|
| |
PiperOrigin-RevId: 211833041
|
|
|
|
| |
PiperOrigin-RevId: 211832421
|
|
|
|
|
|
| |
Test that fetching values of while outputs in sess.run by tensor name works. This tests that an IdentityN node with the same name and outputs as the original while op was added to the graph during lowering.
PiperOrigin-RevId: 211827934
|
|
|
|
| |
PiperOrigin-RevId: 211824645
|
|
|
|
|
|
|
|
|
| |
Previously take_grad represents the average gradients being aggregated.
However this does not cover other use cases such as summing quantiles,
or summing probability distributions from parallel workers. This change
extends the functionality.
PiperOrigin-RevId: 211824519
|
|
|
|
|
|
| |
inference if possible.
PiperOrigin-RevId: 211821596
|
|
|
|
|
|
|
|
| |
Add HloSchedule as a field on HloModule. This will enable scheduling to be a normal HLO pass and enable some passes such as copy insertion to more easily use tighter instruction live ranges based on the schedule. This change required adding HloSchedule to the "hlo" library because of circular dependencies.
Nothing except for tests actually sets the schedule at the moment, but follow up cls will add a scheduling pass which will do so.
PiperOrigin-RevId: 211815293
|
|
|
|
| |
PiperOrigin-RevId: 211813852
|
|
|
|
|
|
| |
RELNOTES: n/a
PiperOrigin-RevId: 211804843
|
|
|
|
|
|
| |
RELNOTES: n/a
PiperOrigin-RevId: 211798892
|
|
|
|
|
|
| |
RELNOTES: n/a
PiperOrigin-RevId: 211798876
|
|
|
|
| |
PiperOrigin-RevId: 211770067
|
|
|
|
| |
PiperOrigin-RevId: 211762464
|
|
|
|
|
|
| |
Profiling showed that we were wastefully (i) heap-allocating and freeing an Executor::Args object on each call, and (as a result) (ii) incurring extra function dispatch overhead in the callback.
PiperOrigin-RevId: 211755493
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
This is so that gradients with respect to scalars pass (see the test added in
backprop_test.py).
A micro benchmark just calling constant_op.constant slows down a bit - this is
inevitable as we are creating a new python object.
After: walltime: ~2.1
Before: walltime: ~1.47
Linear regression benchmark is pretty much unchanged.
PiperOrigin-RevId: 211753801
|
|
|
|
|
|
|
|
| |
is enabled.
Some builds install a `TraceCollector` at process startup, but it is mostly not enabled. This inhibits the recent optimization to avoid accessing `OpKernel::name()` and `OpKernel::type_string()` every time a kernel is launched. By caching the `TraceCollector` in the `TracingDevice` and adding a method to enquire about its state, we increase the applicability of the optimization.
PiperOrigin-RevId: 211752728
|
|
|
|
| |
PiperOrigin-RevId: 211744058
|
|
|
|
| |
PiperOrigin-RevId: 211741560
|
|
|
|
| |
PiperOrigin-RevId: 211733735
|
|
|
|
| |
PiperOrigin-RevId: 211730301
|
|
|
|
|
|
| |
These APIs are based on queue runners, which have been deprecated and will be removed in TensorFlow 2.0. They have been replaced with `tf.data.Dataset`, which provides a more efficient version of the same functionality.
PiperOrigin-RevId: 211727844
|
|
|
|
| |
PiperOrigin-RevId: 211727610
|
|
|
|
|
|
|
|
|
|
|
| |
*** Original change description ***
Add HloSchedule class representing a sequential order of an HloModule.
Currently we represent a sequential schedule of a module using a SequentialHloOrdering::HloModuleSequence which is a type alias of a bare map from HloComputation* to std::vector<HloInstruction*>. This CL replaces this with a proper class which results in better encap...
***
PiperOrigin-RevId: 211726890
|
|
|
|
|
|
|
|
| |
TF 2.0 to return a no-arg callable to output a learning rate, instead of directly outputting a learning rate tensor.
This brings the graph mode API in line with the eager execution API, where this change was made to allow changing the learning rate value across different invocations of optimizer functions.
PiperOrigin-RevId: 211726295
|
|
|
|
| |
PiperOrigin-RevId: 211724843
|
|
|
|
| |
PiperOrigin-RevId: 211722113
|
|
|
|
|
|
| |
Also upload fully quantized mobilenet-v2 and inception-v3 models.
PiperOrigin-RevId: 211721504
|
|
|
|
| |
PiperOrigin-RevId: 211721354
|
|
|
|
| |
PiperOrigin-RevId: 211719399
|
|
|
|
| |
PiperOrigin-RevId: 211719342
|
|
|
|
| |
PiperOrigin-RevId: 211719009
|
|
|
|
|
|
| |
RELNOTES: n/a
PiperOrigin-RevId: 211718248
|
|
|
|
|
|
| |
CancellationManagers for each run of the function.
PiperOrigin-RevId: 211717580
|
|
|
|
| |
PiperOrigin-RevId: 211715608
|
|
|
|
| |
PiperOrigin-RevId: 211714574
|
|
|
|
| |
PiperOrigin-RevId: 211712907
|
|
|
|
| |
PiperOrigin-RevId: 211711493
|
|
|
|
|
|
| |
second axis of Tensors through tf.train.warm_start. Note that the underlying initializer already has this functionality (for example, for output layers).
PiperOrigin-RevId: 211709879
|
|
|
|
|
|
| |
These APIs are based on queue runners, which have been deprecated and will be removed in TensorFlow 2.0. They have been replaced with `tf.data.Dataset`, which provides a more efficient version of the same functionality.
PiperOrigin-RevId: 211708268
|
|
|
|
|
|
|
|
|
| |
There's no need for the two separate classes anymore. This also cleans up some other parts of the interface:
* Removes the clear_resource_control_flow_state, which isn't used anywhere
* Makes capture_value a private method of FuncGraph (_capture_helper)
* Makes create_substitute_placeholder private
PiperOrigin-RevId: 211707906
|
|
|
|
| |
PiperOrigin-RevId: 211707155
|
|
|
|
|
|
| |
implementations of the batch and padded batch reducers.
PiperOrigin-RevId: 211706766
|