aboutsummaryrefslogtreecommitdiffhomepage
Commit message (Collapse)AuthorAge
* Avoid retaining two copies of each constant in `ConstantOp`.Gravatar Derek Murray2018-02-04
| | | | | | | | | | | | | | Presently, the kernel keeps two copies of the constant tensor value, which can be large: 1. In the `ConstantOp::tensor_` field. 2. In the `OpKernel::def_` field (as an attr of the `NodeDef`). Since we can be sure that `ConstantOp` will never need to access the tensor value from `OpKernel::def_`, this change introduces a mechanism for `OpKernel` implementations to store a stripped `NodeDef` in the base class, and uses it in `ConstantOp` to avoid storing the tensor value attr. PiperOrigin-RevId: 184455793
* [XLA] Assign mandatory constraints in a DFS order and non-manatory ↵Gravatar Blake Hechtman2018-02-03
| | | | | | constraints in a BFS order. PiperOrigin-RevId: 184429818
* [tf-signal] Fix exception when input shape is unknown in ↵Gravatar RJ Ryan2018-02-03
| | | | | | mfccs_from_log_mel_spectrograms. PiperOrigin-RevId: 184400783
* [XLA] Add tests for Clamp with scalars S32 and U32.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184376425
* Fix breakage: Can't build TFLite with Bazel for Mac / iOS.Gravatar Yu-Cheng Ling2018-02-02
| | | | PiperOrigin-RevId: 184375534
* A more efficient implementation of the Op using batch operations.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184367562
* Extended TFE_OpSetDevice() with the ability to set an op device from non-GPU ↵Gravatar Mingsheng Hong2018-02-02
| | | | | | | | | back to GPU. Added unit testing, and also refined unit test logic for checking the presence of a GPU device. The latter is needed when we add XLA device support. PiperOrigin-RevId: 184366172
* [XLA] Minor cleanups related to multi-output fusion.Gravatar Justin Lebar2018-02-02
| | | | | | | | | | | | | | | | - Add some comments about preexisting invariants, and add some CHECKs. - In the LoopEmitter constructor, materialize the given ArraySlice<IrArray> to a vector, so we don't rely on the given ArraySlice having any particular lifetime. - Add the invariant that the LoopEmitter constructor which takes a list of IrArrays is only for multi-output fusion. Previously it said: If you only pass one array, then treat it as regular fusion. But this results in an LLVM type mismatch, because the given target_element_generator should be passing a struct with one element. PiperOrigin-RevId: 184365310
* Propagate outfeed sharding, if specified from TensorFlow.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184361221
* Automated g4 rollback of changelist 184347012Gravatar Anna R2018-02-02
| | | | PiperOrigin-RevId: 184360818
* [tf.data] Add public header "tensorflow/core/framework/dataset.h".Gravatar Derek Murray2018-02-02
| | | | | | | | | | | This adds the ability to create a custom C++ Dataset implementation without linking it statically into the TensorFlow library. Note that this internal API is experimental and subject to change between versions of TensorFlow. Fixes #16682. PiperOrigin-RevId: 184356318
* Check that the type of an implicitly dereferenced tensor matches the ↵Gravatar Derek Murray2018-02-02
| | | | | | | | | | expected input type. The dtype of a tensor reference can change between the point when it is "produced" by an operation and consumed by the next operation. This evades checks in the executor that the type of tensor on each edge matches the type signatures of the producing and consuming operation, which could lead to undefined behavior. Although there is no existing operation that changes the type of a tensor reference, it is possible to use the OpKernelContext API to do so, so we add a further check in the runtime to defend against operations that might be added in the future. PiperOrigin-RevId: 184356242
* Update global_step by default if the user specifies a host_call.Gravatar Jonathan Hseu2018-02-02
| | | | PiperOrigin-RevId: 184352399
* Tpu cluster resolver only returns TF server addresses for 'HEALTHY' tpu nodes.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184350480
* Replacing _container_prefix with _graph_key to preserve its use in optimizer ↵Gravatar Alexandre Passos2018-02-02
| | | | | | and deprecate others. PiperOrigin-RevId: 184348303
* TF_CALL_ALL_TYPES should include variantGravatar Alexandre Passos2018-02-02
| | | | PiperOrigin-RevId: 184347081
* Rollback of recent changes to tensor.{h,cc}Gravatar Alexandre Passos2018-02-02
| | | | PiperOrigin-RevId: 184347012
* Adds batch inference support on TPU with TPUEstimator.predict.Gravatar Jianwei Xie2018-02-02
| | | | PiperOrigin-RevId: 184339842
* [TF RNN] Small optimization to rnn: only calculate copy_cond once.Gravatar Eugene Brevdo2018-02-02
| | | | PiperOrigin-RevId: 184335231
* Changed minimal required version of bleach Python package from 1.5 to 2.0. ↵Gravatar Tatiana Shpeisman2018-02-02
| | | | | | bleach 1.5 causes problems with Jupyter as reported in #16424 PiperOrigin-RevId: 184332663
* A few misc tweaks to TFE APIs.Gravatar Mingsheng Hong2018-02-02
| | | | PiperOrigin-RevId: 184329345
* TFLite: Conv CBLAS kernelGravatar Yu-Cheng Ling2018-02-02
| | | | PiperOrigin-RevId: 184328848
* [tf.data] Move framework/dataset.h to framework/dataset_stateful_op_whitelist.h.Gravatar Derek Murray2018-02-02
| | | | | | | | This will make way for the move of kernels/data/dataset.h to framework/dataset.h, while preserving version control history, after which we might recombine the headers. PiperOrigin-RevId: 184327481
* Support `float16` `dtype` in `tf.linalg.*`.Gravatar Joshua V. Dillon2018-02-02
| | | | | | | Note: not all `LinearOperator` functions will support `float16`. This change merely enables constructing the `LinearOperator` object(s) using this `dtype`. PiperOrigin-RevId: 184323477
* Breaking change: Revise HMC interface to accept a list of Tensors ↵Gravatar Joshua V. Dillon2018-02-02
| | | | | | representing a partitioning of chain state. PiperOrigin-RevId: 184323369
* Skip the node that has unexpected number of outputs.Gravatar Yuefeng Zhou2018-02-02
| | | | PiperOrigin-RevId: 184320865
* [TF:XLA] Bump open source llvm revision to r324073Gravatar Sanjoy Das2018-02-02
| | | | PiperOrigin-RevId: 184318036
* Update ops-related pbtxt files.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184317521
* [TF:XLA] Improve/refactor the handling of resource types/shapes.Gravatar Peter Hawkins2018-02-02
| | | | | | | | Previously we used an xla::Shape to track the shape of a resource (Variable, TensorArray, Stack) shape. The xla::Shape described how the resource was represented to XLA, e.g., as a (buffer, size) pair for a Stack resource. Instead, separate the TensorFlow abstract shape representation from the XLA shape representation and track it separately. This leads to simpler and more readable code. PiperOrigin-RevId: 184310694
* Allow ResizeBilinear to resize the output tensor in Prepare(), if the size ↵Gravatar A. Unique TensorFlower2018-02-02
| | | | | | tensor is const. PiperOrigin-RevId: 184309687
* Register resource_scatter_update for string types.Gravatar Alexandre Passos2018-02-02
| | | | PiperOrigin-RevId: 184309674
* Add shape inference for outside_compilation graph rewrite. Pull out enough ↵Gravatar A. Unique TensorFlower2018-02-02
| | | | | | | | | | | | | of the graph to enable inference of the shape of a SendFromHost Op once the shape of corresponding RecvAtHost Ops are known. END_PUBLIC Fixed open source build breaks. BEGIN_PUBLIC Automated g4 rollback of changelist 184169668 PiperOrigin-RevId: 184306845
* Register GPU host kernels for Identity and RefIdentity.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184305574
* Automated g4 rollback of changelist 184273245Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184303789
* Remove hidden_ops.txt file. Instead, switch to use visibility attribute in ↵Gravatar Anna R2018-02-02
| | | | | | ApiDef proto. PiperOrigin-RevId: 184301076
* Consider beyond immediate neighbors to find exit node.Gravatar Jacques Pienaar2018-02-02
| | | | | | | Most of the exit nodes are immediate neighbors of the switch, except we do have cases where the switch feeds into an identity that feeds into a exit. PiperOrigin-RevId: 184297180
* Show that we must over reallocate after resizing dynamic tensors.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184296680
* Fix latent bug in dependency optimizer.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184291701
* Disable graph optimizations (CSE) in test so that constant nodes are not ↵Gravatar Yao Zhang2018-02-02
| | | | | | deduped. PiperOrigin-RevId: 184289685
* [comment-only change]: Fix grammar error.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184285125
* Fix a bug in function inlining when the argument is an implicitly ↵Gravatar Derek Murray2018-02-02
| | | | | | | | dereferenced ref tensor. Previously the inliner would add an identity node with an invalid ref-type attr when the actual parameter had ref type. The changed version removes the reference. PiperOrigin-RevId: 184285084
* Enabling partitioned variables to work with TPU.Gravatar A. Unique TensorFlower2018-02-02
| | | | | | | | | | | | | When partitioned variables are used in a TPU training loop, concat gradient operations get generated for which XLA requires the concat dimension argument to be a constant (or foldable to a constant). However since such constant is defined outside of the train while context an Enter node is generated in order to pass it. The fix consists in detecting such case, and to duplicate the (scalar) constant inside the while context, so that XLA can succesfully process the resulting graph. PiperOrigin-RevId: 184273245
* Fix some tf-lite testsGravatar Gunhan Gulsoy2018-02-02
| | | | PiperOrigin-RevId: 184247187
* Fix tolerance too tight for Wasserstein distance test.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184240222
* Fix some tf-lite testsGravatar Gunhan Gulsoy2018-02-02
| | | | PiperOrigin-RevId: 184247187
* Fix tolerance too tight for Wasserstein distance test.Gravatar A. Unique TensorFlower2018-02-02
| | | | PiperOrigin-RevId: 184240222
* Internal changeGravatar Justin Lebar2018-02-02
| | | | PiperOrigin-RevId: 184239740
* Automated g4 rollback of changelist 183874527Gravatar A. Unique TensorFlower2018-02-01
| | | | PiperOrigin-RevId: 184236409
* Supporting new saving op structure.Gravatar A. Unique TensorFlower2018-02-01
| | | | PiperOrigin-RevId: 184233513
* Add darwin_x86_64 config in TF Lite BUILD file.Gravatar Yu-Cheng Ling2018-02-01
| | | | PiperOrigin-RevId: 184227786
generated by cgit v1.2.3 (git 2.39.1) at 2024-07-06 20:17:38 +0000