OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 00:19:58 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity
119,364 Commits 50 Branches 144 Tags 1.3 GiB
3ce2638c2e
Commit Graph

119364 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Mehdi Amini
3ce2638c2e Internal change 
PiperOrigin-RevId: 403402096
Change-Id: I2d47349c2537808cbb40ca64747e056b21f5648d
 2021-10-15 10:10:41 -07:00
A. Unique TensorFlower
e909039ab1 Update TFRT dependency to use revision 
48d1b66117.

PiperOrigin-RevId: 403393842
Change-Id: Ic7db29c86d422cbc248a38596b8c5323700e8b89
 2021-10-15 09:35:51 -07:00
Peter Hawkins
19f634e469 [XLA:Python] Add missing functions to type stub. 
PiperOrigin-RevId: 403391794
Change-Id: Ia38d268e27f500c00cb1f505d644845ba6fef5ec
 2021-10-15 09:27:12 -07:00
Benjamin Kramer
c39967af21 [XLA] InstructionFusion: Don't fuse across a root instruction 
There may be instructions after the root, but we only need the result of the
root. Don't fuse it with anything that comes below as that would change which
value is returned.

PiperOrigin-RevId: 403381025
Change-Id: If534c77fa5ff236f8fb64c892a6af22c2a4ec362
 2021-10-15 08:36:32 -07:00
A. Unique TensorFlower
0ae96964d1 Go: Update generated wrapper functions for TensorFlow ops. 
PiperOrigin-RevId: 403350971
Change-Id: I8db34bff1743360847138a2d117c653972ca3c36
 2021-10-15 05:52:29 -07:00
Benjamin Kramer
d7aa1eb6e5 [cpurt:jit] Run a canonicalizer pass before lowering to HLO 
This lets the canonicalizer pick up TF-level canonicalizations, like removing
incompatible_shape_error=false on tf.Equal.
PiperOrigin-RevId: 403343803
Change-Id: I4e6e875f95f25efaaebee2706e1c4fa5eee38b43
 2021-10-15 05:04:10 -07:00
Benjamin Kramer
f3287b863c [tf:mhlo] Be more aggressive about flipping incompatible_shape_error to true 
TF shape inference will make this op return a `tensor<*x1>` if it doesn't know
whether the shapes are compatible or not. If we see anything else we know that
the shapes will either be always or never be compatible and adjust the flag
accordingly.

PiperOrigin-RevId: 403318319
Change-Id: I9c19664847904e41441912c5ebb2c1cecc0757f4
 2021-10-15 02:23:38 -07:00
A. Unique TensorFlower
b9506ac185 Update GraphDef version to 920. 
PiperOrigin-RevId: 403315410
Change-Id: I3b0da6eb56d2981b130a3241fd17c1d418f3aaed
 2021-10-15 02:15:03 -07:00
A. Unique TensorFlower
513313e142 compat: Update forward compatibility horizon to 2021-10-15 
PiperOrigin-RevId: 403315379
Change-Id: Ibbe38b9fcd028e16f30aeda2a7b09cefd1f7d5f0
 2021-10-15 02:08:42 -07:00
A. Unique TensorFlower
84a52be7f4 Go: Update generated wrapper functions for TensorFlow ops. 
PiperOrigin-RevId: 403311585
Change-Id: Ia245f9fbb8ea2bfcd9f58655fda870a083c62d1d
 2021-10-15 01:53:29 -07:00
A. Unique TensorFlower
21b2bfa97d Update TFRT dependency to use revision 
f9866013bd.

PiperOrigin-RevId: 403292149
Change-Id: If12e34edfe308c7cc81a6f2204ac1f5060ae4151
 2021-10-14 23:58:38 -07:00
A. Unique TensorFlower
7aaad85825 Go: Update generated wrapper functions for TensorFlow ops. 
PiperOrigin-RevId: 403291767
Change-Id: I827333be407e249226333dd1976226558c2a47f8
 2021-10-14 23:52:43 -07:00
A. Unique TensorFlower
18a37f11ac Update TFRT dependency to use revision 
5cc764dec6.

PiperOrigin-RevId: 403291083
Change-Id: I6ec183c8a13fa3773202bc2c96c4095be4986056
 2021-10-14 23:44:52 -07:00
Yash Katariya
a0522d6847 Docker file for Cuda 11.4 built with Cudnn 8.0.5 
PiperOrigin-RevId: 403283584
Change-Id: I28ebe66bbedd1b4a820d0cac694e8a870f44d4b3
 2021-10-14 22:55:48 -07:00
A. Unique TensorFlower
cf42f72887 Go: Update generated wrapper functions for TensorFlow ops. 
PiperOrigin-RevId: 403273306
Change-Id: I48043313859de719822f6bdcdd5241f59c49c562
 2021-10-14 21:52:02 -07:00
Peter Hawkins
cfc9d3e1ba [XLA:Python] Add an explicit xla_client.pyi interface file. 
This works around a bug in pytype where modules that are re-exported often have the types of their contents replaced with Any, and makes the errors from pytype for xla_client much more useful.

Fix a number of incorrect and missing type annotations, and make a few small cleanups in passing.

PiperOrigin-RevId: 403257225
Change-Id: Ic570f8277691fbdc79ab8d9e516187743b037728
 2021-10-14 20:16:55 -07:00
A. Unique TensorFlower
71e3a6d3a7 Go: Update generated wrapper functions for TensorFlow ops. 
PiperOrigin-RevId: 403254209
Change-Id: I40e99690919eb99e89127274a6fa1df288597950
 2021-10-14 19:54:51 -07:00
Penporn Koanantakool
c79ba87153 Make Transpose's shape inference function validate that negative perm values are within the tensor's rank. 
PiperOrigin-RevId: 403252853
Change-Id: Ia6b31b45b237312668bb31c2c3b3c7bbce2d2610
 2021-10-14 19:42:55 -07:00
Tianrun Li
042dc3be4c Add queue id 
PiperOrigin-RevId: 403247850
Change-Id: I453d93cf0c4c9ce10aab82e002a24b34bd001acc
 2021-10-14 19:11:34 -07:00
Diego Caballero
2a1a976c6d [tfrt:jit] Disable 'Any' reduction test. 
The test is flaky. It needs further investigation.

PiperOrigin-RevId: 403240419
Change-Id: If4f74420a424693512be2038ec0e6f2071f29509
 2021-10-14 18:33:05 -07:00
A. Unique TensorFlower
44a5e24577 Go: Update generated wrapper functions for TensorFlow ops. 
PiperOrigin-RevId: 403233112
Change-Id: I254d625018571eab4e346067f4fd74ada71501d2
 2021-10-14 17:51:03 -07:00
Haoyu Zhang
e8002b0aca Introduce CoordinationServiceConfig to make the coordination service more configurable. 
PiperOrigin-RevId: 403221941
Change-Id: I73baebecc941edf3b3a9bfe003b8db34f6c47945
 2021-10-14 16:59:30 -07:00
Roman Dzhabarov
a91435160c Replace std::unordered_map with the abs::flat_hash_map in the ResourceManager. 
PiperOrigin-RevId: 403214130
Change-Id: I18ec3db3823a9e99ef0121696036092d3de1e4f7
 2021-10-14 16:33:54 -07:00
Peng Wang
fbb6388c75 Fixes nightly breakage by falling back from a fast path to a slow path in _assertAllCloseRecursive when NotImplementedError is raised. Also rolls back a previous fix on a upper level. 
PiperOrigin-RevId: 403213148
Change-Id: I7e94aeb729d6a2351121156f8621e6f5d29e3d67
 2021-10-14 16:28:15 -07:00
David Majnemer
364752fdfc [XLA] Check all reduce/reduce-window operands 
PiperOrigin-RevId: 403212187
Change-Id: Ie44992af6861c6434cb5c64f635eeaafa2222c4f
 2021-10-14 16:20:47 -07:00
Scott Zhu
48cc691d3d Remove the legacy directory of tensorflow/python/keras/estimator 
PiperOrigin-RevId: 403208301
Change-Id: Ib790b8241d2d0f337175fe3caad321b0b124b9d7
 2021-10-14 16:06:21 -07:00
Stella Laurenzo
1690fa8e1c Use the new mlir_configure_python_dev_packages() macro instead of open coded. 
This is required for consistency across upstream/downstream as there are certain deployment environments that cannot resolve the 'Development' component.

PiperOrigin-RevId: 403207802
Change-Id: Ibb47c8c616c0135c0568f7166de6528c239f9998
 2021-10-14 16:00:18 -07:00
A. Unique TensorFlower
cd6d20bb8f Report errors from individual GPU conv autotuning failures. 
PiperOrigin-RevId: 403201665
Change-Id: Id0325181cafc42542e0c4b76bf1e4f187ebdc37f
 2021-10-14 15:31:52 -07:00
A. Unique TensorFlower
30eeb202e3 Update TFRT dependency to use revision 
0579df0f20.

PiperOrigin-RevId: 403196321
Change-Id: Id006b2cfbcacab32b079aa8f5f79d72290d6c01c
 2021-10-14 15:22:03 -07:00
George Karpenkov
ec73eb2099 [XLA:GPU] Bump down verbosity level for LLVM diagnostic handler 
VLOG(1) seems overly verbose

PiperOrigin-RevId: 403196139
Change-Id: Ideb86249abb87e47074296c552c05c4583a17ce4
 2021-10-14 15:15:55 -07:00
A. Unique TensorFlower
0f84381b22 Add xla_prefer_single_graph_cluster field to ConfigProto.Experimental to express preference for how XLA clusters the graph. 
PiperOrigin-RevId: 403194061
Change-Id: Iee933298701066cc919fe4fd6b63697bc2e8298d
 2021-10-14 15:11:04 -07:00
Faizan Muhammad
483ac0b5bb Introduce TraceType for Iterator 
PiperOrigin-RevId: 403194058
Change-Id: Ieda3c876f2e18bc19e8cecc23e1725ec1c146d48
 2021-10-14 15:04:27 -07:00
Xiao Yu
23260248e9 Support HostMemory annotation in KernelFallbackOpHandler. TF uses HostMemory to annotate host input and output. After this change, KernelFallbackOpHandler should be able to execute ALL gpu ops and tpu eager ops properly. 
PiperOrigin-RevId: 403193565
Change-Id: Ic2c9c395d77e1394b8c2e7e255105478d6e49385
 2021-10-14 14:59:01 -07:00
Blake Hechtman
08fcfaed8f [XLA:ALGEBRAIC_SIMPLIFIER] cleanup the IsConvertPairNoOp to use the primitive util formulation. 
PiperOrigin-RevId: 403189355
Change-Id: I4efa6af1de823b96601731459d816384b2a3ebe0
 2021-10-14 14:43:36 -07:00
A. Unique TensorFlower
7574ff260c Update TFRT dependency to use revision 
23956998f8.

PiperOrigin-RevId: 403177553
Change-Id: I1479b5b0e3b4b886942e35fe4fb250ee23d79f57
 2021-10-14 13:52:30 -07:00
Kuangyuan Chen
d1d9afda4d Remove unnecessary calls of WorkQueueInterface::InitializeRequest() in TFRT GPU 
PiperOrigin-RevId: 403176042
Change-Id: Id8c093033c11534400d0fb40564e7c442465d64b
 2021-10-14 13:47:26 -07:00
A. Unique TensorFlower
ecc1fc7be3 Integrate LLVM at llvm/llvm-project@fcbec7e668 
Updates LLVM usage to match
[fcbec7e668ec](https://github.com/llvm/llvm-project/commit/fcbec7e668ec)

PiperOrigin-RevId: 403175209
Change-Id: I0b5d40721dbabf8281451afa4f103b8350b95545
 2021-10-14 13:42:14 -07:00
Faizan Muhammad
ac8b31f8c2 Normalize Protocol Goldens Generation 
PiperOrigin-RevId: 403171391
Change-Id: I486aa30488c6fa90a56fa3939cc2bb9bb277ff80
 2021-10-14 13:27:26 -07:00
A. Unique TensorFlower
c3da9e2f00 Report errors from individual GPU conv autotuning failures. 
PiperOrigin-RevId: 403170540
Change-Id: I2cea725644ead824f4f06c148c5dcbb2b3e865c7
 2021-10-14 13:21:35 -07:00
Penporn Koanantakool
a813a102b2 Update oneDNN to version 2.4.1 to fix XByak-induced crashes on non-Intel systems. Details here: 
ad901e5489

PiperOrigin-RevId: 403166809
Change-Id: I31d17e43ad08c175a9326cd3e2daa38899a077fb
 2021-10-14 13:06:22 -07:00
TensorFlower Gardener
90eeb59514 Merge pull request #52358 from Intel-tensorflow:gyshi/test_replace_add_with_biasadd 
PiperOrigin-RevId: 403163981
Change-Id: If883ac2cba2f91224226eb1057198c00324b6561
 2021-10-14 12:58:15 -07:00
Jose Baiocchi
de5192ffc8 Remove kernel annotation from xplane schema 
PiperOrigin-RevId: 403162482
Change-Id: Ie00b06abbf9d6987be0933b128ae75eeefcfe933
 2021-10-14 12:51:20 -07:00
Michael Delorimier
16f7a2f1d8 Proper failures for unsupported unranked and dynamic shapes in ConvertInplaceUpdateOp. Previously an assertion was failing. 
PiperOrigin-RevId: 403160273
Change-Id: I68773aef5da0ac88f39a8c708d23d101284399d7
 2021-10-14 12:45:16 -07:00
A. Unique TensorFlower
2803656187 Update TFRT dependency to use revision 
08402be5a3.

PiperOrigin-RevId: 403159616
Change-Id: I4b60dd9fbb84c9e553a7cfdf3d1ad7715477ccc0
 2021-10-14 12:40:37 -07:00
A. Unique TensorFlower
7ea556628c Get rid of absl::flat_hash_map to fix the Windows build. 
It's good that there was reason to revisit this code anyway, since it looks
like there was a bug of comparing iterators across unrelated maps in the
checked-in version.

PiperOrigin-RevId: 403158423
Change-Id: I7236c2e6f3ba25934ac7438c00a493e5948db0bf
 2021-10-14 12:34:47 -07:00
Jake VanderPlas
be30c80eb2 fix repr() of jit-compiled functions 
PiperOrigin-RevId: 403157400
Change-Id: I1d7ec24aeacc2dfe555f39dde0a7d0faf6468791
 2021-10-14 12:29:24 -07:00
TensorFlower Gardener
562a230495 Merge pull request #52374 from rsuderman:PadV2 
PiperOrigin-RevId: 403151220
Change-Id: Ic8b094b789e6bfd2f235568a5e2ef3f397ea7095
 2021-10-14 12:08:16 -07:00
A. Unique TensorFlower
24f01dfaac Update TFRT dependency to use revision 
bb80b98ff1.

PiperOrigin-RevId: 403129005
Change-Id: I530503d729b5764442af33ab994af9ec670fb43a
 2021-10-14 10:39:58 -07:00
A. Unique TensorFlower
95fd0007e1 Integrate LLVM at llvm/llvm-project@25a3463c44 
Updates LLVM usage to match
[25a3463c44ee](https://github.com/llvm/llvm-project/commit/25a3463c44ee)

PiperOrigin-RevId: 403125985
Change-Id: I79e83d0acd82ae74374a99808229d98c093d8db8
 2021-10-14 10:30:06 -07:00
Eugene Zhulenev
36dac265e4 [tf][tfrt] Add tests for numerical correctness of tf.math operations 
A set of tests that verify that tf.math operations produce identical results when they are compiled or computed with the regular Tensorflow kernels.

PiperOrigin-RevId: 403124206
Change-Id: If6553758b5f104b557370f348d2bc8eeca88f7f2
 2021-10-14 10:24:43 -07:00