| Commit message (Collapse) | Author | Age |
... | |
|
|
|
|
|
|
|
| |
This allows us to avoid repeated index lookups and perform a sequential scan of the index in the common case where we are doing a full restore, or a restore from a sub-model. It also dramatically reduces excessive restore parallelism.
Testing with a checkpoint with 1000 100x100 tensors, restoring from CNS drops from ~1m to ~5 seconds.
PiperOrigin-RevId: 180827583
|
|
|
|
| |
PiperOrigin-RevId: 180825516
|
|
|
|
|
|
| |
C-API or python API are returning the same types--mostly the string and unicode types.
PiperOrigin-RevId: 180821956
|
|
|
|
| |
PiperOrigin-RevId: 180820115
|
|
|
|
| |
PiperOrigin-RevId: 180818654
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
And update version to 1.5.0-rc0.
Starting with 1.5.0-rc0, it will be possible to use GPUs from TensorFlow Java
programs by adding the following to the application's pom.xml:
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>libtensorflow</artifactId>
<version>1.5.0-rc0</version>
</dependency>
<dependency>
<groupId>org.tensorflow</groupId>
<artifactId>libtensorflow_jni_gpu</artifactId>
<version>1.5.0-rc0</version>
</dependency>
Updates #12909
PiperOrigin-RevId: 180816468
|
|
|
|
|
|
| |
export location in bundle-shim.
PiperOrigin-RevId: 180814251
|
|
|
|
| |
PiperOrigin-RevId: 180811282
|
|
|
|
| |
PiperOrigin-RevId: 180809736
|
|
|
|
| |
PiperOrigin-RevId: 180809175
|
|
|
|
| |
PiperOrigin-RevId: 180806246
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
knows nothing about other modules. Therefore, sends and recvs are
given a default layout to ensure that the layout is the same across
all communicating modules.
This CL adds a ChannelLayoutConstraints object which can be passed
into LayoutAssignment. When a send or recv is layed out, its chosen
layout is saved into the ChannelLayoutConstraints. When, later,
another instruction is seen in a different module that uses the same
communicating channel ID, the layout is constrained to be the same as
in the previous module.
PiperOrigin-RevId: 180771799
|
|
|
|
| |
PiperOrigin-RevId: 180762948
|
|
|
|
| |
PiperOrigin-RevId: 180755908
|
|
|
|
| |
PiperOrigin-RevId: 180754374
|
|
|
|
|
|
|
|
|
|
|
|
| |
This change teaches XLA to maintain cycle counters specifically for non-entry
computations, like computations representing the body of a While. Without this
change, instructions in such non-entry computations are noted as taking 0.00% of
their execution time which isn't ideal.
Implementation-wise, this just falls out of uniformly using a
std::unordered_map<T, int64> for both the HloInstruction->ProfileIndex and the
HloComputation->ProfileIndex mappings.
PiperOrigin-RevId: 180750463
|
|
|
|
|
|
| |
inference.
PiperOrigin-RevId: 180750305
|
|
|
|
| |
PiperOrigin-RevId: 180749642
|
|
|
|
|
|
|
| |
from TF runtime to debug server, using
TensorBoardDebugWrapperSession and TensorBoardDebugHook.
PiperOrigin-RevId: 180748754
|
|
|
|
| |
PiperOrigin-RevId: 180747698
|
|
|
|
|
|
| |
The "reserved" keyword doesn't work in enumerations, replace it with a comment.
PiperOrigin-RevId: 180747499
|
|
|
|
| |
PiperOrigin-RevId: 180746153
|
|
|
|
| |
PiperOrigin-RevId: 180744498
|
|
|
|
|
|
|
|
|
|
|
| |
DotOpEmitter
- Move VectorSupportLibrary to under service/cpu since it is specific to the
CPU backend.
- Use TargetMachineFeatures to infer the vector width in DotOpEmitter
- Move the kAvxVectorSize magic constant into TargetMachineFeatures
PiperOrigin-RevId: 180740693
|
|
|
|
| |
PiperOrigin-RevId: 180739249
|
|
|
|
| |
PiperOrigin-RevId: 180738639
|
|
|
|
|
|
|
|
|
| |
section.
This op (not in the public API yet) allows exclusive access to certain resources
by bottlenecking subgraph access to run within a serialized critical section.
PiperOrigin-RevId: 180733901
|
|
|
|
|
|
|
|
| |
Prior to this change, entering an init_scope while in graph mode with an empty
graph stack would enable eager execution, if eager execution had previously
been enabled; this wasn't the desired behavior.
PiperOrigin-RevId: 180730994
|
|
|
|
|
|
|
|
|
|
|
|
| |
Specifically, make control_flow_util.GetOutputContext robust to
imported exit nodes (which don't have control flow contexts). This was
a bug prior to the C API being enabled in that imported exit nodes
would not have contexts, but it happened to not be exposed. Note that
importing a metagraph will add the contexts back after doing the
initial import, but there's still a window where no contexts are
assigned.
PiperOrigin-RevId: 180730785
|
|
|
|
| |
PiperOrigin-RevId: 180730491
|
|
|
|
|
|
|
|
|
|
| |
Clients should ideally prevent such functions from being created in
the first place, but we still want the runtime to be robust to
malformed functions. Trying to run functions with invalid control flow
constructs can result in crashes or hangs, so we want to catch it
before running.
PiperOrigin-RevId: 180727589
|
|
|
|
|
|
|
| |
RNG_BERNOULLI is easy to compose out of other operations and appears to provide
no real benefit. Let's remove it.
PiperOrigin-RevId: 180726889
|
|
|
|
|
|
| |
different input batch size. Fixes #14985.
PiperOrigin-RevId: 180724096
|
|
|
|
| |
PiperOrigin-RevId: 180721390
|
|
|
|
|
|
|
|
|
| |
This invariant is incorrect; the parameter will never be the shape of the use's
first operand; in fact the param should match operand(1)'s shape for
DynamicSlice and operand(2)'s shape for DynamicUpdateSlice (since start_indices is the argument that we are trying to constrain the bounds for). This is checked
for in CreateLiteralForConstrainedUses().
PiperOrigin-RevId: 180717508
|
|
|
|
| |
PiperOrigin-RevId: 180705266
|
|
|
|
|
|
| |
Also includes a pretty printer API.
PiperOrigin-RevId: 180698491
|
|
|
|
| |
PiperOrigin-RevId: 180691955
|
|
|
|
|
|
|
| |
Formatting was also cleaned up, without making any schema or
documentation changes.
PiperOrigin-RevId: 180688094
|
|
|
|
| |
PiperOrigin-RevId: 180679688
|
|
|
|
| |
PiperOrigin-RevId: 180677538
|
|
|
|
|
|
|
|
|
| |
This change also puts the necessary build infrastructure in place so we can
continue to build many proper lightweight SQLite extensions in the future that
are easy for users to copy over into their own environments, so they can
readily access their data.
PiperOrigin-RevId: 180673039
|
|
|
|
|
|
| |
kfac.loss_functions_lib._allowed_symbols.
PiperOrigin-RevId: 180672608
|
|
|
|
| |
PiperOrigin-RevId: 180671590
|
|
|
|
| |
PiperOrigin-RevId: 180670333
|
|
|
|
| |
PiperOrigin-RevId: 180669193
|
|
|
|
|
|
|
|
| |
This makes the output much more consistent.
Also round the probabilities to 2 decimal places.
PiperOrigin-RevId: 180666095
|
|
|
|
|
|
| |
See #15326
PiperOrigin-RevId: 180645977
|
|
|
|
|
|
| |
order. This made shape inference far less efficient than it should be in some cases.
PiperOrigin-RevId: 180629882
|
|
|
|
| |
PiperOrigin-RevId: 180628481
|