aboutsummaryrefslogtreecommitdiffhomepage
Commit message (Collapse)AuthorAge
...
* | | | | | | | | Merge pull request #22386 from girving:statelessGravatar TensorFlower Gardener2018-10-05
|\ \ \ \ \ \ \ \ \ | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215995215
* | | | | | | | | | Simply the logic for bubbling captured tensors when building cond_v2 grad.Gravatar Saurabh Saxena2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | The current logic tries to bubble the forward pass tensor to the outermost graph. That might not always be do-able e.g. when the cond is inside a while loop it will need to know accumulator logic for while_loop. So instead, the cond_grad now captures tensors from the forward If op's graph. When the grad If op is built these tensors will be appropriately captured by the surrounding FuncGraph. PiperOrigin-RevId: 215993009
* | | | | | | | | | [XLA:GPU] Remove hidden flag for disabling heuristic layout assignment.Gravatar Justin Lebar2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | Heuristic NCHW/NHWC layout assignment works great; we've never had to flip this flag. Might as well remove it and simplify things a bit. PiperOrigin-RevId: 215989807
* | | | | | | | | | [tf.data vectorization] Feed inputs to vectorizers with notion of stackednessGravatar Rachel Lim2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215989259
* | | | | | | | | | Automated rollback of commit d258207f1583df4faa452265b051879af6c15dacGravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215989111
* | | | | | | | | | [XLA:GPU] Use a struct for the return value of ↵Gravatar Justin Lebar2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | CudnnConvolutionAlgorithmPicker::PickBestAlgorithm. Using a struct lets us return additional data -- namely, the elapsed time to run the best algo -- without adding a fourth entry to the tuple, which would be confusing. No functional change. PiperOrigin-RevId: 215987795
* | | | | | | | | | Fix api_compatibility_test diff for large files. assertEqual might be appliedGravatar Anna R2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | instead of assertMultiLineEqual if input is too large (https://bugs.python.org/issue11763). This change is switching to use unified_diff in that case. PiperOrigin-RevId: 215987656
* | | | | | | | | | Orders non-resource-affecting stateful ops in defuns.Gravatar Alexandre Passos2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215985679
* | | | | | | | | | Merge pull request #21183 from AndreasMadsen:sparsemax-2Gravatar TensorFlower Gardener2018-10-05
|\ \ \ \ \ \ \ \ \ \ | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215981773
* | | | | | | | | | | [TF:XLA] Bump open source abseil revision to ↵Gravatar Sanjoy Das2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | e821380d69a549dc64900693942789d21aa4df5e PiperOrigin-RevId: 215981413
* | | | | | | | | | | Internal changeGravatar Jared Duke2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215978771
* | | | | | | | | | | Removes the INTEL_MKL_ML_ONLY option from the CMakeLists build file since ↵Gravatar Penporn Koanantakool2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | the main logic for INTEL_MKL_ML_ONLY is getting removed in PR#22783. #22783 PiperOrigin-RevId: 215978712
* | | | | | | | | | | Add DistributionStrategy support to moving average APIs.Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | Fixes #21405. PiperOrigin-RevId: 215973401
| | * | | | | | | | | Expand stateless random generators to match their stateful cousinsGravatar Geoffrey Irving2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | stateless_random_uniform now take minval+maxval and handles ints, and stateless_normal/stateless_truncated_normal take mean+stddev. Additionally, all of the stateless functions now have proper doc strings. This is step one of moving stateless random numbers out of contrib.
* | | | | | | | | | | Add deprecation call-out for tf_mobileGravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215971335
* | | | | | | | | | | Automated rollback of commit ae0bc6f006497cc04a2ee75166d4ec71c7154fd8Gravatar Jiri Simsa2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215969360
* | | | | | | | | | | Fix bug in nonpip builds in ci_parameterized_build.shGravatar Todd Wang2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | The extra spaces were confusing bash's string-line-continuation from the backslash `\` on the previous line. PiperOrigin-RevId: 215964853
* | | | | | | | | | | [tf.data] Adding specialization for `MapDataset`, `ParallelMapDataset`, and ↵Gravatar Jiri Simsa2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | `MapAndBatchDataset` whose user-provided functions have the property that each output argument take its value directly from an input argument (e.g. `lambda x, y: y, x`). This specialization can produce the result without having to schedule the function using the executor. PiperOrigin-RevId: 215957592
* | | | | | | | | | | Remove commented out code errantly checked in.Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215957544
* | | | | | | | | | | Handle Range & BatchMatMul in partial Flex modeGravatar Yu-Cheng Ling2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215957535
* | | | | | | | | | | [XLA] Use the highest possible precision for large Iota inputs.Gravatar Blake Hechtman2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215957327
* | | | | | | | | | | Copy device from If op to the lowered ops.Gravatar Saurabh Saxena2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | Enable GPU tests for cond_v2. PiperOrigin-RevId: 215956220
* | | | | | | | | | | Break up build --define <option_name>=true into two steps:Gravatar Yifei Feng2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | 1) define bazel config build:<bazel_config_name> --define <option_name>s=true 2) set the config build --config=<bazel_config_name> PiperOrigin-RevId: 215951614
* | | | | | | | | | | Internal change.Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215951354
* | | | | | | | | | | Workaround build errors in Android NDK r14b.Gravatar Jeremy Lau2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215950376
* | | | | | | | | | | Brings V2 Optimizers into Keras w/ Keras signaturesGravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215950207
* | | | | | | | | | | Convert TensorFlow's jpeg dependency to new third party import method.Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215948571
* | | | | | | | | | | Merge pull request #20476 from yongtang:06052018-bincount-shapeGravatar TensorFlower Gardener2018-10-05
|\ \ \ \ \ \ \ \ \ \ \ | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215947463
* | | | | | | | | | | | Profiler collects the number of replicas and num cores per replica used in ↵Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | the model. PiperOrigin-RevId: 215947354
* | | | | | | | | | | | Revert constant folding to previous state.Gravatar Tong Shen2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215946205
* | | | | | | | | | | | Do 2 warmup runs in assert_no_new_pyobjects_executing_eagerly.Gravatar Todd Wang2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215944829
* | | | | | | | | | | | [XLA] Extend the HLO verifier to check that non-layout-changing instructionsGravatar Bixia Zheng2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | preserve operand layouts. Add an std::function member to the HloVerifier for a backend to specify the function object used to determine whether an instruction can change layouts. Use the function object to find out the non-layout-changing instructions and check that such instructions should produce results with the same layouts as its operands. Add test cases. PiperOrigin-RevId: 215941282
* | | | | | | | | | | | Disable micro/examples/micro_speech:micro_speech_test test under msanGravatar Smit Hinsu2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215939542
* | | | | | | | | | | | Make gradient tape stack thread localGravatar Igor Ganichev2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215937618
* | | | | | | | | | | | Declare that stateless random ops are not differentiable in C++ code.Gravatar Tong Shen2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215935319
| | * | | | | | | | | | make sparsemax nan and infinity safeGravatar Andreas Madsen2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | logits that are -inf will be given 0 probability and logits that are inf will result in a nan output. Likewise if all logits are -inf the output will also be nan. This is done by using where operators, mostly because 0 * inf = nan and x/0 = sign(x) inf following the IEEE 754 standard. However these results are not mathematically correct in the context of the sparsemax algorithm. Fixes: https://github.com/tensorflow/tensorflow/issues/15564
* | | | | | | | | | | | Fix documentation.Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215930596
* | | | | | | | | | | | Refactoring TFLite export code. Unify OperatorCode generation logic.Gravatar Yu-Cheng Ling2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215928419
* | | | | | | | | | | | When running a native/builtin op via eager C API, automatically fill in defaultGravatar Mingsheng Hong2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | attr values that are not overridden e.g. transpose_a in the matmul op). This is required for backward compatibility (a binary built via an older version of TF should still run on a newer version of TF, where some ops may have added attrs). For non-eager graph building, the default attr values of graph ops are added by tensorflow::AddDefaultsToNodeDef(). We ran into this issue when running the same S4TF test cases via eager APIs -- some tests failed due to "missing attrs", but are fixed by this patch. PiperOrigin-RevId: 215927271
* | | | | | | | | | | | - Don't set tpu optimizer parameter variable during weight initialization if ↵Gravatar Jing Li2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | the optimizer isn't set, e.g. loading weights and then predict. - Add load_weights for `KerasTpuModel`. PiperOrigin-RevId: 215920993
* | | | | | | | | | | | Update XlaSort to match the underlying HLO.Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215917470
* | | | | | | | | | | | BEGIN_PUBLICGravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | Automated rollback of PR #21945 END_PUBLIC Automated rollback of commit 863f61412fcc654840c6b67473b742ea4e5e964e. Revert #21945. PiperOrigin-RevId: 215913175
* | | | | | | | | | | | Switch NCCL to build from open source (version 2.3.5-5) by default.Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | Note to users manually patching ptxas from a later toolkit version: Building NCCL requires the same version of ptxas and nvlink. PiperOrigin-RevId: 215911973
* | | | | | | | | | | | Relax some unnecessary 4D array restrictionsGravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215910400
* | | | | | | | | | | | [TF:XLA] Move broadcasting code out of BroadcastTo op into a common helper ↵Gravatar Peter Hawkins2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | library. Change XlaBinaryOp::Broadcast to use the BroadcastTo lowering, since it produces fewer extraneous reshapes and transposes. Even if the reshapes and transposes would later optimize away, this yields more readable output and makes life easier for HLO rewrites that run early. Change in preparation for removing reshapes from SoftmaxCrossEntropyWithLogits. PiperOrigin-RevId: 215906847
* | | | | | | | | | | | Use absl::Span for HloModuleGroupMetadataGravatar HyoukJoong Lee2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215905026
* | | | | | | | | | | | compat: Update forward compatibility horizon to 2018-10-05Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215874612
* | | | | | | | | | | | Merge the different LSTM EvalFloat/EvalHybrid calls into a single file.Gravatar A. Unique TensorFlower2018-10-05
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215870962
* | | | | | | | | | | | Remove dependency on epsilon for diagonal shampoo.Gravatar A. Unique TensorFlower2018-10-04
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215857772
* | | | | | | | | | | | Fix link in eager notebook stub.Gravatar Billy Lamberta2018-10-04
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | PiperOrigin-RevId: 215853105