| Commit message (Collapse) | Author | Age |
... | |
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215073584
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215048726
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
Currently, we run tests on machines with GPUs based on the "gpu" tag, and the
tests automatically adapt to whether a GPU is available. Creating two targets,
one tagged with "gpu" and one not, will make us run the tests in both modes.
PiperOrigin-RevId: 215045035
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
of tf.cond, which allows the static shape to be propagated correctly when available.
PiperOrigin-RevId: 215034102
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
learning rate to be modified at runtime. The implementation is not yet complete.
PiperOrigin-RevId: 215030536
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215029224
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215027511
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215026418
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
(actual implementation is pending).
Added comments with pointers to C++ implementations of optimizers.
PiperOrigin-RevId: 215026002
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215025019
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215021487
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215020524
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
kernel supports any combination of the dtypes, and does not need a separate kernel for each dtype.
PiperOrigin-RevId: 215019812
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215019058
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215018984
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215016968
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215016286
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215015490
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215014737
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215011835
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
48cd2c3f351ff188bc85684b84a91b6e6d17d896
This has absl::flat_hash_map in it.
PiperOrigin-RevId: 215011713
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215010842
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215009955
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
{Lookup,Create,LookupOrCreate}Resource().
PiperOrigin-RevId: 215008650
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215005698
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
core (and added that as a base class for all the contrib tests). Also changed the assertDatasetsEqual functions so they are both graph and eager compatible (took the code from CSVDatasetTest) :)
PiperOrigin-RevId: 215004892
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 215003704
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
Since we now have support for FeatureColumnV2 for both DNN and Linear models,
adding tests for the combined canned estimators as well.
PiperOrigin-RevId: 215002573
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 214997213
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
summaries are written at the correct interval for jobs with long-running
evaluations.
PiperOrigin-RevId: 214993119
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
the duration of a single RunInternal() call from RunHandlerPool. It is used for
running inter-op closures with a global scheduler (which in the future) to
improve both median and tail latency (for use-cases like CPU inference).
In the case that global pools aren't used, this change should be a no-op.
PiperOrigin-RevId: 214992852
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 214989908
|
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | |
| | | | | |
Without this the default toolchain is used for a subset of the build and the
tests do not actually run on GPUs.
This uncovered a setup problem in the Docker image that needed fixing.
PiperOrigin-RevId: 214987676
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 214986756
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 214986255
|
| | | | |
| | | | |
| | | | |
| | | | | |
PiperOrigin-RevId: 214985873
|
|\ \ \ \ \
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 214983237
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
Remove dead logical branch.
PiperOrigin-RevId: 214980627
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
Add a single test flag for enabling v2 control flow in tests since we do not plan to support v2 ops with legacy control flow.
We have 2 test decorators now:
@with_control_flow_v2: Enables all tests in a class to run with v2 control flow.
@disable_control_flow_v2: Disables a test function from running in v2. I have removed the skiptests to avoid setup/teardown overheads.
Enable tests in control_flow_ops_py_test that run with control_flow_v2.
PiperOrigin-RevId: 214980108
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 214974535
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 214967868
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 214964988
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 214964640
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
with associated stats_aggregator.
PiperOrigin-RevId: 214961678
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
We will re-enable it when it is more robust.
PiperOrigin-RevId: 214956066
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 214952666
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
dependency
only to the sparse operators that need it.
Automated rollback of commit 120620caf23a044b8aa2db6ba5984384ec936009
PiperOrigin-RevId: 214950946
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 214949709
|
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
Fixes an issue where losses created while executing eagerly were returned as unevaluated lambdas in a defun.
Lazily evaluates Layer losses by default when possible. Even when graph building this is generally a better thing to do (e.g. losses called in a while_loop).
Allows calls to Layer.add_loss when executing eagerly, but only for losses which are not conditional on inputs (no activity regularizers).
PiperOrigin-RevId: 214947108
|
| | | | | |
| | | | | |
| | | | | |
| | | | | | |
PiperOrigin-RevId: 214946257
|