platform/upstream/tensorflow.git
6 years agoComments and little improvements to documentation parser
karl@kubx.ca [Tue, 24 Apr 2018 13:09:01 +0000 (09:09 -0400)]
Comments and little improvements to documentation parser

6 years agoImprove Javadoc and include first code review
karl@kubx.ca [Fri, 6 Apr 2018 12:56:54 +0000 (08:56 -0400)]
Improve Javadoc and include first code review

6 years agoParse op definition and generate a Java Op class.
Karl Lessard [Wed, 28 Mar 2018 21:28:16 +0000 (05:28 +0800)]
Parse op definition and generate a Java Op class.

6 years agoFix minor typos (#19070)
Sergii Khomenko [Fri, 4 May 2018 02:22:09 +0000 (04:22 +0200)]
Fix minor typos (#19070)

6 years agoUpdate learning.py (#19064)
Aditya Yogi [Thu, 3 May 2018 18:13:40 +0000 (23:43 +0530)]
Update learning.py (#19064)

6 years agoFixed Typos (#18806)
Nicholas Nadeau, P.Eng., AVS [Thu, 3 May 2018 17:47:06 +0000 (13:47 -0400)]
Fixed Typos (#18806)

* fixed typos

6 years agoGrammar fixes on architecture.md (#19035)
ctiijima [Thu, 3 May 2018 15:34:07 +0000 (08:34 -0700)]
Grammar fixes on architecture.md (#19035)

6 years agoMerge pull request #19040 from tensorflow/merge_branch
Amit Patankar [Thu, 3 May 2018 14:18:04 +0000 (07:18 -0700)]
Merge pull request #19040 from tensorflow/merge_branch

Merge r1.8 back to master after 1.8.0 release.

6 years agocorrect code snippets to python3 style (#19052)
Letian Feng [Thu, 3 May 2018 13:46:16 +0000 (15:46 +0200)]
correct code snippets to python3 style (#19052)

6 years agoEnabling support for S3 and Google Storage for the MKL Docker image. (#19039)
Elson Rodriguez [Thu, 3 May 2018 04:58:44 +0000 (21:58 -0700)]
Enabling support for S3 and Google Storage for the MKL Docker image. (#19039)

6 years agoFix typos (#18475)
Taehoon Lee [Thu, 3 May 2018 03:01:04 +0000 (12:01 +0900)]
Fix typos (#18475)

6 years agoAdd go_package to proto definition files (#17262)
Yong Tang [Thu, 3 May 2018 00:45:29 +0000 (02:45 +0200)]
Add go_package to proto definition files (#17262)

* Add go_package to proto definition files

This fix tries to address the issue raised in 16282 by
add go_package to proto files, so that generated go files
have correct path.

This fix fixes 16282.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add go_package to proto definition in tensorflow/core/framework

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add go_package to proto definition in tensorflow/core/example

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add go_package to proto definition in tensorflow/core/example

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
6 years agoMerge pull request #18909 from samikama/optimization_pass
gracehoney [Wed, 2 May 2018 22:50:03 +0000 (15:50 -0700)]
Merge pull request #18909 from samikama/optimization_pass

Optimization pass and Memory allocator integration

6 years agoMerge 1.8.0 back to master.
Yifei Feng [Wed, 2 May 2018 22:24:50 +0000 (15:24 -0700)]
Merge 1.8.0 back to master.

6 years agoFix breaking tests
gracehoney [Wed, 2 May 2018 21:46:12 +0000 (14:46 -0700)]
Fix breaking tests

6 years agoMerge pull request #19030 from annarev/copy_modules_before_iterating
annarev [Wed, 2 May 2018 21:45:53 +0000 (14:45 -0700)]
Merge pull request #19030 from annarev/copy_modules_before_iterating

Copy module list before iterating over it

6 years agoCopy module list file in the while statement instead of creating a new variable
Anna R [Wed, 2 May 2018 20:44:32 +0000 (13:44 -0700)]
Copy module list file in the while statement instead of creating a new variable

6 years agoMerge pull request #19031 from smitshilu/patch-3
Amit Patankar [Wed, 2 May 2018 20:07:37 +0000 (13:07 -0700)]
Merge pull request #19031 from smitshilu/patch-3

command Typo

6 years agocommand Typo
Smit Shilu [Wed, 2 May 2018 19:23:31 +0000 (15:23 -0400)]
command Typo

6 years agoCopy module list before iterating over it. Also, import python module for clarity
Anna R [Wed, 2 May 2018 18:38:44 +0000 (11:38 -0700)]
Copy module list before iterating over it. Also, import python module for clarity

6 years agoFix formatting and linter issues
gracehoney [Wed, 2 May 2018 18:21:02 +0000 (11:21 -0700)]
Fix formatting and linter issues

6 years agoFix Makefile to not use benchmark anymore (switch to minimal) (#19019)
Andrew Selle [Wed, 2 May 2018 17:33:07 +0000 (10:33 -0700)]
Fix Makefile to not use benchmark anymore (switch to minimal) (#19019)

Minimal uses nothing and does almost nothing, but it does nothing
requiring protos or rest of tensorflow runtime.

Benchmark_model originally was more like this, but it became
useful for actually benchmarking, making it less useful as a minimal
example.

6 years agoMerge pull request #19025 from caisq/branch_195061425
Shanqing Cai [Wed, 2 May 2018 17:32:47 +0000 (13:32 -0400)]
Merge pull request #19025 from caisq/branch_195061425

Branch 195061425

6 years agoMerge pull request #18989 from ctiijima/fix-benchmarks.md
Amit Patankar [Wed, 2 May 2018 15:53:34 +0000 (08:53 -0700)]
Merge pull request #18989 from ctiijima/fix-benchmarks.md

Grammar fixes in BENCHMARKS.md

6 years agoMerge branch 'master' of https://github.com/tensorflow/tensorflow into optimization_pass
gracehoney [Wed, 2 May 2018 15:52:39 +0000 (08:52 -0700)]
Merge branch 'master' of https://github.com/tensorflow/tensorflow into optimization_pass

6 years agoAllow tfdbg mouse down scroll in curses UI (#18942)
Paul Van Eck [Wed, 2 May 2018 14:08:30 +0000 (07:08 -0700)]
Allow tfdbg mouse down scroll in curses UI (#18942)

* Allow tfdbg mouse down scroll in curses UI

This commit allows users to continuously scroll the screen when the mouse
is held down on the scroll bar when using the curses UI.

* Only allow click-hold scrolling on scroll bar arrows

6 years agoReverting changes from 495d511 that break install_pip_packages.sh in Ubuntu 16.04...
Clayne Robison [Wed, 2 May 2018 13:47:58 +0000 (06:47 -0700)]
Reverting changes from 495d511 that break install_pip_packages.sh in Ubuntu 16.04 containers, causing nightly mkl ci builds to fail. (#18888)

6 years agoFix string issue for temp_export_dir (#18951)
Sergii Khomenko [Wed, 2 May 2018 13:39:46 +0000 (15:39 +0200)]
Fix string issue for temp_export_dir (#18951)

* Fix string issue for temp_export_dir

6 years agoMerge commit for internal changes
Shanqing Cai [Wed, 2 May 2018 13:36:14 +0000 (09:36 -0400)]
Merge commit for internal changes

6 years agoBUGFIX: Convert inputs and list of gradients into tuple if they are not instance...
A. Unique TensorFlower [Wed, 2 May 2018 08:36:18 +0000 (01:36 -0700)]
BUGFIX: Convert inputs and list of gradients into tuple if they are not instance of tuple. Otherwise this causes "unhashable keys" error when we try to hash.
Also fixed lint error.

PiperOrigin-RevId: 195061425

6 years agoAdd missing colocated element in test in buffer_assignment_test.
Mark Heffernan [Wed, 2 May 2018 03:14:29 +0000 (20:14 -0700)]
Add missing colocated element in test in buffer_assignment_test.
This was resulting in a flaky test because sometimes the live set would include this missing colocated element perhaps because the buffers in the allocation has some nondeterministic order (read from a map?).

PiperOrigin-RevId: 195039311

6 years agoMinor refactor: establish some operator naming conventions and apply them, so that...
Dan Moldovan [Wed, 2 May 2018 02:05:39 +0000 (19:05 -0700)]
Minor refactor: establish some operator naming conventions and apply them, so that the interface is a bit more consistent.

PiperOrigin-RevId: 195034691

6 years agoMake the CRF work when sequence_lengths are int32.
Patrick Nguyen [Wed, 2 May 2018 02:02:10 +0000 (19:02 -0700)]
Make the CRF work when sequence_lengths are int32.

PiperOrigin-RevId: 195034218

6 years ago[XLA:CPU] Re-use the same llvm::GlobalVariable for identical literals
Sanjoy Das [Wed, 2 May 2018 01:46:31 +0000 (18:46 -0700)]
[XLA:CPU] Re-use the same llvm::GlobalVariable for identical literals

This isn't necessary today, but it will be after an optimization change I'm
about to make.

LLVM has a constant merging pass too, but one of the motivations here is to
avoid the LLVM compile time overhead of having many large arrays in the IR.

PiperOrigin-RevId: 195032900

6 years agoInternal change
A. Unique TensorFlower [Wed, 2 May 2018 00:59:59 +0000 (17:59 -0700)]
Internal change

PiperOrigin-RevId: 195028221

6 years agoInternal change.
A. Unique TensorFlower [Wed, 2 May 2018 00:57:02 +0000 (17:57 -0700)]
Internal change.

PiperOrigin-RevId: 195027918

6 years agoRe-apply CL 194140820, which reverts #18251 (convolution change).
Patrick Nguyen [Wed, 2 May 2018 00:48:36 +0000 (17:48 -0700)]
Re-apply CL 194140820, which reverts #18251 (convolution change).

PiperOrigin-RevId: 195027049

6 years agotest fix
Mustafa Ispir [Wed, 2 May 2018 00:21:24 +0000 (17:21 -0700)]
test fix

PiperOrigin-RevId: 195023740

6 years agoMerge pull request #18730 from joel-shor/master
Joel Shor [Wed, 2 May 2018 00:08:57 +0000 (03:08 +0300)]
Merge pull request #18730 from joel-shor/master

[tf.data] Check in a strictly faster rejection resampling transformation.

6 years agoSharding for tensorflow/contrib/timeseries/python/timeseries/state_space_models:struc...
Allen Lavoie [Tue, 1 May 2018 23:53:51 +0000 (16:53 -0700)]
Sharding for tensorflow/contrib/timeseries/python/timeseries/state_space_models:structural_ensemble_test

PiperOrigin-RevId: 195019968

6 years agoFix wrongly ordered lines
Rachel Lim [Tue, 1 May 2018 23:52:24 +0000 (16:52 -0700)]
Fix wrongly ordered lines

PiperOrigin-RevId: 195019769

6 years agoAvoid making a copy of the graph needlessly
Benoit Steiner [Tue, 1 May 2018 23:38:23 +0000 (16:38 -0700)]
Avoid making a copy of the graph needlessly

PiperOrigin-RevId: 195017837

6 years agoAdds logistic_regression_head.
A. Unique TensorFlower [Tue, 1 May 2018 23:38:19 +0000 (16:38 -0700)]
Adds logistic_regression_head.

PiperOrigin-RevId: 195017830

6 years agoCheck for overflow in shape calculation.
A. Unique TensorFlower [Tue, 1 May 2018 23:33:03 +0000 (16:33 -0700)]
Check for overflow in shape calculation.

PiperOrigin-RevId: 195017114

6 years agoAllow `warm_start_from` argument to be a SavedModel path.
A. Unique TensorFlower [Tue, 1 May 2018 23:20:47 +0000 (16:20 -0700)]
Allow `warm_start_from` argument to be a SavedModel path.

PiperOrigin-RevId: 195015356

6 years ago[TF:XLA] Separate on-host and on-device shape and layout in HloModule.
A. Unique TensorFlower [Tue, 1 May 2018 23:17:21 +0000 (16:17 -0700)]
[TF:XLA] Separate on-host and on-device shape and layout in HloModule.

Previously, only one layout was stored with an HLO module. This CL allows
HLO passes to modify the on-device layouts without affecting the on-host
layout (provided by the client)

PiperOrigin-RevId: 195014875

6 years ago[tf.data] Remove debug code.
joel-shor [Tue, 1 May 2018 23:14:30 +0000 (02:14 +0300)]
[tf.data] Remove debug code.

6 years ago[tf.data] Try fixing the Windows build by adding the directed interleave kernel to...
joel-shor [Tue, 1 May 2018 23:01:14 +0000 (02:01 +0300)]
[tf.data] Try fixing the Windows build by adding the directed interleave kernel to this cmake file.

6 years agoImplementation of the fully-connected TFLite Op using the symmetric quantization.
A. Unique TensorFlower [Tue, 1 May 2018 22:47:27 +0000 (15:47 -0700)]
Implementation of the fully-connected TFLite Op using the symmetric quantization.

PiperOrigin-RevId: 195010312

6 years agoGo: Update generated wrapper functions for TensorFlow ops.
A. Unique TensorFlower [Tue, 1 May 2018 22:47:26 +0000 (15:47 -0700)]
Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 195010310

6 years agoFixed some outdated comments
Benoit Steiner [Tue, 1 May 2018 22:19:42 +0000 (15:19 -0700)]
Fixed some outdated comments

PiperOrigin-RevId: 195006088

6 years agoMinor JNI performance improvement.
A. Unique TensorFlower [Tue, 1 May 2018 22:01:22 +0000 (15:01 -0700)]
Minor JNI performance improvement.

PiperOrigin-RevId: 195002949

6 years agoMaking ids unique in nn.embedding_lookup_sparse. This helps to reduce RPC calls for...
A. Unique TensorFlower [Tue, 1 May 2018 22:00:20 +0000 (15:00 -0700)]
Making ids unique in nn.embedding_lookup_sparse. This helps to reduce RPC calls for looking up the embeddings when there are repeated ids in the batch.

PiperOrigin-RevId: 195002785

6 years ago[tf.data] Fix debug output.
joel-shor [Tue, 1 May 2018 21:56:43 +0000 (00:56 +0300)]
[tf.data] Fix debug output.

6 years agoFix bug in peak buffer accounting in buffer assignment.
Mark Heffernan [Tue, 1 May 2018 21:52:40 +0000 (14:52 -0700)]
Fix bug in peak buffer accounting in buffer assignment.
Buffer assignment keeps track of the set of logical buffers which are live at
the point of peak memory usage for each allocation. Previously colocated
buffers were not properly accounted for. This CL addresses this problem.

PiperOrigin-RevId: 195001567

6 years agoMerge changes from github.
Patrick Nguyen [Tue, 1 May 2018 21:28:36 +0000 (14:28 -0700)]
Merge changes from github.

PiperOrigin-RevId: 194997009

6 years agoMake tower-local variables non-trainable even with the default
A. Unique TensorFlower [Tue, 1 May 2018 21:27:33 +0000 (14:27 -0700)]
Make tower-local variables non-trainable even with the default
DistributionStrategy.

PiperOrigin-RevId: 194996819

6 years agoEnable checkpointless eval and predict for tf.estimator.
A. Unique TensorFlower [Tue, 1 May 2018 21:04:59 +0000 (14:04 -0700)]
Enable checkpointless eval and predict for tf.estimator.

PiperOrigin-RevId: 194993191

6 years agoUpdate community/swift
Mark Daoust [Tue, 1 May 2018 20:54:34 +0000 (13:54 -0700)]
Update community/swift

PiperOrigin-RevId: 194991305

6 years agoUpdate schema.
Shashi Shekhar [Tue, 1 May 2018 20:44:58 +0000 (13:44 -0700)]
Update schema.

PiperOrigin-RevId: 194989704

6 years agoRelax the stringent memory allocator constraints in AssignOp if a Grappler graph...
A. Unique TensorFlower [Tue, 1 May 2018 20:34:39 +0000 (13:34 -0700)]
Relax the stringent memory allocator constraints in AssignOp if a Grappler graph analysis determines it to be safe. This will allow Assign to reuse the input buffer to initialize the variable in many cases.

PiperOrigin-RevId: 194988134

6 years agoCollective Ops Part 5
A. Unique TensorFlower [Tue, 1 May 2018 20:15:53 +0000 (13:15 -0700)]
Collective Ops Part 5

Distributed-mode implementations of DeviceResolverInterface
and ParamResolverInterface.  Extend Worker interface with
new methods in support of these interfaces.

This change is part of a series of changes introducing infrastructure
for collective ops and initial implementations of reduction and broadcast.

PiperOrigin-RevId: 194984585

6 years agoOpen source infeed test
Sanjoy Das [Tue, 1 May 2018 20:07:03 +0000 (13:07 -0700)]
Open source infeed test

PiperOrigin-RevId: 194983270

6 years agoAdd utility to auto shard a dataset pipeline in the appropriate place by locating...
Priya Gupta [Tue, 1 May 2018 20:01:41 +0000 (13:01 -0700)]
Add utility to auto shard a dataset pipeline in the appropriate place by locating the file readers and sharding their input files.

PiperOrigin-RevId: 194982311

6 years agoAdd a pointer from Device to its owning DeviceMgr.
Peter Hawkins [Tue, 1 May 2018 19:56:29 +0000 (12:56 -0700)]
Add a pointer from Device to its owning DeviceMgr.
Allow remote function execution on TPU devices.

PiperOrigin-RevId: 194981511

6 years agoReview updates
Sami Kama [Tue, 1 May 2018 19:54:04 +0000 (12:54 -0700)]
Review updates

6 years agoImplements matrix multiply-accumulate for linear no-offset (aka symmetric) quantizer.
A. Unique TensorFlower [Tue, 1 May 2018 19:39:52 +0000 (12:39 -0700)]
Implements matrix multiply-accumulate for linear no-offset (aka symmetric) quantizer.

PiperOrigin-RevId: 194978865

6 years agoAdd device_util.resolve method which merges with current device as well.
Yuefeng Zhou [Tue, 1 May 2018 19:24:38 +0000 (12:24 -0700)]
Add device_util.resolve method which merges with current device as well.

PiperOrigin-RevId: 194976633

6 years agoSimplified shape inference.
Benoit Steiner [Tue, 1 May 2018 19:17:48 +0000 (12:17 -0700)]
Simplified shape inference.

PiperOrigin-RevId: 194975603

6 years agoImprove shape inference for tf.contrib.signal.frame.
RJ Ryan [Tue, 1 May 2018 19:02:59 +0000 (12:02 -0700)]
Improve shape inference for tf.contrib.signal.frame.

PiperOrigin-RevId: 194972934

6 years agoBoosted trees: support indicator column.
A. Unique TensorFlower [Tue, 1 May 2018 18:52:43 +0000 (11:52 -0700)]
Boosted trees: support indicator column.

PiperOrigin-RevId: 194971229

6 years agoeager: Update sample notebooks with API changes in the last few releases.
Asim Shankar [Tue, 1 May 2018 18:52:04 +0000 (11:52 -0700)]
eager: Update sample notebooks with API changes in the last few releases.

Most notably:
- Avoid using tf.contrib.eager since equivalent functionality if available outside tf.contrib
- Datasets can be directly iterated on.
- Use tf.GradientTape instead of tf.contrib.eager.implicit_gradients

PiperOrigin-RevId: 194971115

6 years agoAutomated g4 rollback of changelist 194917415
Gunhan Gulsoy [Tue, 1 May 2018 18:04:26 +0000 (11:04 -0700)]
Automated g4 rollback of changelist 194917415

PiperOrigin-RevId: 194962702

6 years ago[tf.data] Fix debug output.
joel-shor [Tue, 1 May 2018 16:53:30 +0000 (19:53 +0300)]
[tf.data] Fix debug output.

6 years agoFix crash in HloGraphDumper where it crashes on tuple shaped constants
A. Unique TensorFlower [Tue, 1 May 2018 16:07:57 +0000 (09:07 -0700)]
Fix crash in HloGraphDumper where it crashes on tuple shaped constants

The problem is that it tries to use a special logic for 0 element constants
but the logic used to check the number of elements only supports array shapes.

PiperOrigin-RevId: 194945246

6 years ago[tf.data] Fix BUILD file.
joel-shor [Tue, 1 May 2018 15:58:55 +0000 (18:58 +0300)]
[tf.data] Fix BUILD file.

6 years ago[tf.data] More debug code, since the previous 'fix' wasn't a fix.
joel-shor [Tue, 1 May 2018 15:35:33 +0000 (18:35 +0300)]
[tf.data] More debug code, since the previous 'fix' wasn't a fix.

6 years ago[tf.data] More debug code, since the previous 'fix' wasn't a fix.
joel-shor [Tue, 1 May 2018 15:33:18 +0000 (18:33 +0300)]
[tf.data] More debug code, since the previous 'fix' wasn't a fix.

6 years agoPreventing RemoveTrivialBinary from removing broadcasts.
A. Unique TensorFlower [Tue, 1 May 2018 14:56:08 +0000 (07:56 -0700)]
Preventing RemoveTrivialBinary from removing broadcasts.

PiperOrigin-RevId: 194937001

6 years agoProtocol buffer classes now list their fields in dir(cls)
A. Unique TensorFlower [Tue, 1 May 2018 10:25:26 +0000 (03:25 -0700)]
Protocol buffer classes now list their fields in dir(cls)

PiperOrigin-RevId: 194917415

6 years ago[XLA:CPU] Open source some tests.
Sanjoy Das [Tue, 1 May 2018 07:42:56 +0000 (00:42 -0700)]
[XLA:CPU] Open source some tests.

PiperOrigin-RevId: 194903752

6 years ago Remove whitespace characters from tf_cuda_compute_capabilities user string (#18986)
Maciej [Tue, 1 May 2018 03:30:58 +0000 (22:30 -0500)]
 Remove whitespace characters from tf_cuda_compute_capabilities user string (#18986)

Remove all whitespace characters from the user specified tf_cuda_compute_capabilities string as this can results in errors during the split operation, and is easy for users to do as it is natural to insert a space after a comma

6 years agoFix MSVC openmp flag (#18973)
Ben [Tue, 1 May 2018 03:30:22 +0000 (23:30 -0400)]
Fix MSVC openmp flag (#18973)

* Fix MSVC openmp flag

6 years agoFixes for review comments
Sami Kama [Tue, 1 May 2018 03:22:51 +0000 (20:22 -0700)]
Fixes for review comments

6 years agoUpdate ops-related pbtxt files.
A. Unique TensorFlower [Tue, 1 May 2018 02:18:40 +0000 (19:18 -0700)]
Update ops-related pbtxt files.

PiperOrigin-RevId: 194883351

6 years agoMove LinearOperatorKronecker and LinearOperatorBlockDiag to core.
A. Unique TensorFlower [Tue, 1 May 2018 01:51:48 +0000 (18:51 -0700)]
Move LinearOperatorKronecker and LinearOperatorBlockDiag to core.

PiperOrigin-RevId: 194881237

6 years ago[XLA] Redesign: dump HloSnapshot at the point where it used to dump the SessionModule.
A. Unique TensorFlower [Tue, 1 May 2018 01:41:36 +0000 (18:41 -0700)]
[XLA] Redesign: dump HloSnapshot at the point where it used to dump the SessionModule.

PiperOrigin-RevId: 194880385

6 years agoInternal change.
A. Unique TensorFlower [Tue, 1 May 2018 01:05:37 +0000 (18:05 -0700)]
Internal change.

PiperOrigin-RevId: 194877173

6 years agoRemove proto header import from core/framework/tracking_allocator.h
Yifei Feng [Tue, 1 May 2018 01:01:23 +0000 (18:01 -0700)]
Remove proto header import from core/framework/tracking_allocator.h

The goal is to make kernels mostly independent of proto headers, which will let us lock down our .so import.

PiperOrigin-RevId: 194876569

6 years agoGo: Update generated wrapper functions for TensorFlow ops.
A. Unique TensorFlower [Tue, 1 May 2018 00:46:35 +0000 (17:46 -0700)]
Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 194874988

6 years ago[XLA] Redesign: migrate tensorflow/compiler/tf2xla, tensorflow/compiler/aot:
A. Unique TensorFlower [Tue, 1 May 2018 00:41:33 +0000 (17:41 -0700)]
[XLA] Redesign: migrate tensorflow/compiler/tf2xla, tensorflow/compiler/aot:
- xla::ComputationBuilder -> xla::XlaBuilder
- xla::ComputationDataHandle -> xla::XlaOp
- xla::Computation -> xla::XlaComputation
- xla::CompileOnlyClient::AotComputationInstance -> xla::CompileOnlyClient::AotXlaComputationInstance
- xla::SessionModule -> xla::HloSnapshot

PiperOrigin-RevId: 194874462

6 years ago[tf.data] Adding an experimental `group_by_reducer` transformation which groups eleme...
Jiri Simsa [Tue, 1 May 2018 00:38:38 +0000 (17:38 -0700)]
[tf.data] Adding an experimental `group_by_reducer` transformation which groups elements of an input pipeline by a key, applies a reduce function to elements of each group "on-the-fly", and outputs the results once all input elements have been processed.

PiperOrigin-RevId: 194874087

6 years agoInternal cleanup.
A. Unique TensorFlower [Tue, 1 May 2018 00:14:50 +0000 (17:14 -0700)]
Internal cleanup.

PiperOrigin-RevId: 194871141

6 years agoFaster reduce_logsoftmax (specially in eager) and bugfixes in broadcast_to
Alexandre Passos [Tue, 1 May 2018 00:11:40 +0000 (17:11 -0700)]
Faster reduce_logsoftmax (specially in eager) and bugfixes in broadcast_to

PiperOrigin-RevId: 194870645

6 years agoSmall fix to prevent a crash if the delegate has not implemented FreeBufferHandle.
A. Unique TensorFlower [Mon, 30 Apr 2018 23:43:14 +0000 (16:43 -0700)]
Small fix to prevent a crash if the delegate has not implemented FreeBufferHandle.

PiperOrigin-RevId: 194866595

6 years agoAdd MultiNodeDataset and MultiNodeIterator which are intended to work for multi-node...
Yuefeng Zhou [Mon, 30 Apr 2018 23:12:33 +0000 (16:12 -0700)]
Add MultiNodeDataset and MultiNodeIterator which are intended to work for multi-node distribution strategy.

PiperOrigin-RevId: 194862215

6 years ago[XLA] Change the TF2XLA bridge to perform F16 reduction using F32 data type.
Bixia Zheng [Mon, 30 Apr 2018 23:11:38 +0000 (16:11 -0700)]
[XLA] Change the TF2XLA bridge to perform F16 reduction using F32 data type.

Add test cases to test that reduce sum for bfloat16 and float16 doesn't lose too
much precision.

PiperOrigin-RevId: 194862078

6 years agoImprove error message for pip_smoke_test.
Yifei Feng [Mon, 30 Apr 2018 22:56:26 +0000 (15:56 -0700)]
Improve error message for pip_smoke_test.

PiperOrigin-RevId: 194859591

6 years agoDo not allocate memory for literal as it will be allocated later.
A. Unique TensorFlower [Mon, 30 Apr 2018 22:41:28 +0000 (15:41 -0700)]
Do not allocate memory for literal as it will be allocated later.

PiperOrigin-RevId: 194857422

6 years agoEnhancements to GRAPHVIZ_DOT output:
A. Unique TensorFlower [Mon, 30 Apr 2018 22:19:00 +0000 (15:19 -0700)]
Enhancements to GRAPHVIZ_DOT output:
    -edge weights added to encourage straighter main data-flow
    -line thickness proportional to log(data_size)
    -set global parameter "nslimit" to prevent excessive layout time for difficult graphs

PiperOrigin-RevId: 194854051