| Commit message (Collapse) | Author | Age |
|
|
|
|
|
|
|
|
| |
Generalize the static analysis across while and for loops.
Convert len builtin to tf.shape()[0].
Add for loop canonicalization and companion tests.
Modify the template behavior for Name nodes to let the template control the target, which allows simplifying the caller.
PiperOrigin-RevId: 181633983
|
|
|
|
|
|
| |
This partially addresses the concerns in issue #13161.
PiperOrigin-RevId: 181629980
|
|
|
|
| |
PiperOrigin-RevId: 181629745
|
|
|
|
| |
PiperOrigin-RevId: 181625889
|
|
|
|
|
|
| |
Passing around copies of std::functions incurs heap allocations and deallocations, which, unfortunately, matters in this case. Minimize the amount of copies.
PiperOrigin-RevId: 181625079
|
|
|
|
|
|
| |
Changing the default value of colocate_gradients_with_ops to True.
PiperOrigin-RevId: 181624864
|
|
|
|
|
|
| |
tf.AUTO_REUSE type.
PiperOrigin-RevId: 181620379
|
|
|
|
| |
PiperOrigin-RevId: 181619109
|
|
|
|
|
|
|
| |
This allows LLVM to vectorize loads/stores in these kernels, among other
things.
PiperOrigin-RevId: 181618991
|
|
|
|
|
|
| |
definition
PiperOrigin-RevId: 181617501
|
|
|
|
| |
PiperOrigin-RevId: 181610422
|
|
|
|
|
|
|
|
| |
Remove obsolete shape_size_fn_ from HloVerifier/ShapeVerifier.
Adds a rank check to FFT shape inference.
PiperOrigin-RevId: 181601294
|
|
|
|
|
|
|
|
|
| |
Note that there are already existing checks in BufferAllocation::AddAssignment
to ensure all buffers are no larger than the allocation. But colocated buffer
sets are used to handle forced aliasing, e.g. kWhile, kCall and kConditional,
which require all buffers to be identically sized.
PiperOrigin-RevId: 181565074
|
|
|
|
|
|
| |
each other.
PiperOrigin-RevId: 181563474
|
|
|
|
| |
PiperOrigin-RevId: 181553949
|
|
|
|
|
|
| |
Started home page.
PiperOrigin-RevId: 181548668
|
|
|
|
| |
PiperOrigin-RevId: 181548597
|
|
|
|
| |
PiperOrigin-RevId: 181548023
|
|
|
|
| |
PiperOrigin-RevId: 181547286
|
|
|
|
|
|
| |
Shapes were not correctly converted to C types.
PiperOrigin-RevId: 181546120
|
|
|
|
|
|
|
|
| |
Also enables the C API in slot_create_test.py, which exercises the new
behavior (previously it would fail because it would create an op with
no OpDef and list inputs).
PiperOrigin-RevId: 181544033
|
|
|
|
| |
PiperOrigin-RevId: 181537854
|
|
|
|
|
|
| |
have been used.
PiperOrigin-RevId: 181532901
|
|
|
|
|
|
| |
CPU and 10% faster on GPU.
PiperOrigin-RevId: 181528321
|
|
|
|
| |
PiperOrigin-RevId: 181527872
|
|
|
|
|
|
|
|
| |
Bazel silently uses the wrong build settings for --config=android_arm64 (--cpu=arm64-v8a is not enough), and actually still uses armeabi-v7a in at least some cases. --fat_apk_cpu fixes this.
See #15581 for more.
PiperOrigin-RevId: 181525260
|
|
|
|
|
|
| |
to fix toolchain installation error
PiperOrigin-RevId: 181524891
|
|
|
|
|
|
| |
Fixes #15737
PiperOrigin-RevId: 181523430
|
|
|
|
| |
PiperOrigin-RevId: 181523204
|
|
|
|
|
|
|
| |
Merge input values at phi nodes correctly: If a phi operand is the phi itself,
and the other operands are all the same, then the phi node is redundant.
PiperOrigin-RevId: 181521522
|
|
|
|
| |
PiperOrigin-RevId: 181519635
|
|
|
|
|
|
| |
ExecutionProfile::compute_cycle_count never worked for CPU and GPU with Hlo
profiling disabled, as far as I can tell.
PiperOrigin-RevId: 181517824
|
|
|
|
|
|
| |
needs to use the scope symbols, not their last assigned value.
PiperOrigin-RevId: 181511978
|
|
|
|
| |
PiperOrigin-RevId: 181511871
|
|
|
|
| |
PiperOrigin-RevId: 181511142
|
|
|
|
|
|
|
|
| |
Runtime constant folding after the graph has been rewritten to include any
feeds, so it's safe and desirable to constant fold PlaceholderWithDefaults
at this point.
PiperOrigin-RevId: 181510650
|
|
|
|
| |
PiperOrigin-RevId: 181508517
|
|
|
|
| |
PiperOrigin-RevId: 181506626
|
|
|
|
|
|
|
| |
* Nesting is implemented by sharing a single EagerVariableStore among a top-level EagerTemplate and all children EagerTemplate objects that are nested underneath it. Variables added to an EagerTemplate object are also added to all EagerTemplate objects under which it is nested.
* This change also simplifies the implementation of __call__ for both Template and EagerTemplate.
PiperOrigin-RevId: 181506600
|
|
|
|
|
|
| |
MacOS build fails for missing include of <array>
PiperOrigin-RevId: 181506335
|
|
|
|
| |
PiperOrigin-RevId: 181505090
|
|
|
|
|
|
| |
both client and server side. Thread count is hardcoded to 8 for now, should be tuned in the future.
PiperOrigin-RevId: 181504374
|
|
|
|
|
|
| |
easier to package custom ops (tfmini) with the core binary on iOS.
PiperOrigin-RevId: 181503662
|
|
|
|
|
|
| |
For example, if you have defined a namedtuple called `MyNamedTuple`, and there are two variables `a=MyNamedTuple(...)`, and `b=MyNamedTuple(...)`, you can directly call `assertAllClose(a, b)` if you intend to know if the two namedtuples are close elementwise.
PiperOrigin-RevId: 181501832
|
|
|
|
|
|
|
|
|
|
|
|
| |
These fusion categories are really just a way of expressing a particular
kind of dot or conv. This makes them easier to differentiate from
"proper" fusion nodes.
We also change the category of these instructions so that in the HLO
profile, e.g. conv-fusion shows up under the convolution category,
rather than under "fusion".
PiperOrigin-RevId: 181499300
|
|
|
|
| |
PiperOrigin-RevId: 181494416
|
|
|
|
| |
PiperOrigin-RevId: 181494232
|
|
|
|
|
|
|
|
|
|
|
| |
* Previously, strong assumptions were made about how numpy.ndarrays
are formatted as strings. This led to breakages due to certain
unclear changes in numpy or its dependencies. This CL relaxes the
assumption and fix the affected tests for tfdbg and eager.
* The tests in tensor_format_test.py are simplified through helper
methods.
PiperOrigin-RevId: 181494182
|
|
|
|
| |
PiperOrigin-RevId: 181493377
|
|
|
|
|
|
| |
types whitelisted to remain uncompiled.
PiperOrigin-RevId: 181493349
|