aboutsummaryrefslogtreecommitdiffhomepage
Commit message (Collapse)AuthorAge
* [XLA:GPU] Add an implementation of scatter for GPUGravatar Benjamin Kramer2018-10-09
| | | | | | | | | | | | This simple has a kernel that runs on every element of the updates tensor, figure out the right indices to perform the update, and applies it with an atomic operation. Currently we emit a CAS for plain (i.e. non-add) updates, which is inefficient. Also TuplePointsToAnalysis doesn't know that it should alias the operand and output buffers of a scatter, which would avoid a copy. PiperOrigin-RevId: 216412467
* Small cleanup in function_test.Gravatar Gunhan Gulsoy2018-10-09
| | | | PiperOrigin-RevId: 216412380
* Update ops-related pbtxt files.Gravatar A. Unique TensorFlower2018-10-09
| | | | PiperOrigin-RevId: 216410913
* Add RaggedTensors to tf.core. Moving the RaggedGather op kernel.Gravatar A. Unique TensorFlower2018-10-09
| | | | PiperOrigin-RevId: 216400726
* Silence tf.distributions deprecation messages caused by internal global ↵Gravatar Pavel Sountsov2018-10-09
| | | | | | | | | | | | function calls. E.g. register_kl calls would trigger such warnings. This spam was exacerbated by the fact that it happens before logging is initialized, so it is dumped prominently to STDERR. Worse yet it also happened no matter whether the user imported any symbols from tf.distributions or not as the relevant code is executed when you import TensorFlow. PiperOrigin-RevId: 216396036
* [tf.data] NUMA-aware MapAndBatch dataset.Gravatar Brennan Saeta2018-10-09
| | | | PiperOrigin-RevId: 216395709
* Return ::tensorflow::Status in Toco Graph Transformations.Gravatar Yu-Cheng Ling2018-10-09
| | | | PiperOrigin-RevId: 216392908
* Update ops-related pbtxt files.Gravatar A. Unique TensorFlower2018-10-09
| | | | PiperOrigin-RevId: 216392772
* Improves tf.function prototype.Gravatar Alexandre Passos2018-10-09
| | | | | | | | | | Specifically: - renames from def_function - returns an object with well-defined methods - doesn't force-retrace twice - uses the python descriptor API ( https://docs.python.org/3/howto/descriptor.html ) to remove the need for a tf.method PiperOrigin-RevId: 216388957
* Update TFLite Converter documentation.Gravatar Nupur Garg2018-10-09
| | | | PiperOrigin-RevId: 216386450
* Internal change.Gravatar Nupur Garg2018-10-09
| | | | PiperOrigin-RevId: 216385202
* Removing the _SHOULD_RECORD_SUMMARIES_NAME andGravatar Rohan Jain2018-10-09
| | | | | | | _SUMMARY_WRITER_INIT_COLLECTION_NAME collections from the summaryV2 implementation. Replacing them with global variables. PiperOrigin-RevId: 216383152
* [tf.data vectorization] Handle captured inputs in MapVectorization optimizationGravatar Rachel Lim2018-10-09
| | | | PiperOrigin-RevId: 216381943
* Fixes typo in Sort description.Gravatar Tayo Oguntebi2018-10-09
| | | | PiperOrigin-RevId: 216375421
* Go: Update generated wrapper functions for TensorFlow ops.Gravatar A. Unique TensorFlower2018-10-09
| | | | PiperOrigin-RevId: 216372144
* Include live-in symbols in liveness analysis. These are required for control ↵Gravatar Dan Moldovan2018-10-09
| | | | | | flow conversion. PiperOrigin-RevId: 216370439
* Improve docstring for tf.data.Dataset.shuffle()Gravatar A. Unique TensorFlower2018-10-09
| | | | PiperOrigin-RevId: 216370329
* Create SDCAOptimizerV2 op to fix the "adaptative" typo.Gravatar Yuefeng Zhou2018-10-09
| | | | PiperOrigin-RevId: 216370193
* Correctly pre-reserve visit state in HloInstruction::PostOrderDFSGravatar A. Unique TensorFlower2018-10-09
| | | | | | | | | | Previously we pre-reserverd the visit state based on the number of instructions but then started to index it with the instruction unique ID what can be larger then the instruction count. This resulted in some very expensive re-allocations what can be eliminated by reserving the correctly sized buffer. PiperOrigin-RevId: 216369849
* Change LOG(WARNING) to VLOG(1) in utilsGravatar Peter Ma2018-10-09
| | | | PiperOrigin-RevId: 216369081
* Throw error when evaluating have variable target in GradientTape.Gravatar Tamara Norman2018-10-09
| | | | PiperOrigin-RevId: 216368178
* Internal changeGravatar Jared Duke2018-10-09
| | | | PiperOrigin-RevId: 216367867
* Avoid extra calls to set_random_seed, as it is already called inGravatar Gunhan Gulsoy2018-10-09
| | | | | | tensorflowtestcase. PiperOrigin-RevId: 216363450
* Allowing for mixture of V1 and V2 feature columns usage in canned ↵Gravatar Rohan Jain2018-10-09
| | | | | | | | | | estimators. This is required for TF hub use cases where users might send in new feature columns to old model code. Implemented this support by making V2 feature columns support the V1 API. This is needed temporarily and would definitely be removed by TF 2.0, possibly earlier depending on what guarantees are provided by TF hub. The only case we don't allow here is mixing in V2 shared embedding columns with V1 Feature columns. V2 Shared FC's depend on a SharedEmbeddingState manager that would have to be passed in to the various API's and there wasn't really a very clean way to make that work. Mixing V2 feature columns with V1 shared embedding columns is fine though and along with all other combinations PiperOrigin-RevId: 216359041
* Fixing Toco for exporting graphs with stringsGravatar A. Unique TensorFlower2018-10-09
| | | | | | | If the graph contains not constant array with strings it fails because the array's size can't be estimated. PiperOrigin-RevId: 216356162
* Removed unused load statements from the core BUILD.Gravatar A. Unique TensorFlower2018-10-09
| | | | PiperOrigin-RevId: 216354906
* Automated rollback of commit 375c109659d2d0e6265447dffdeb460693b3cccfGravatar A. Unique TensorFlower2018-10-09
| | | | PiperOrigin-RevId: 216350134
* compat: Update forward compatibility horizon to 2018-10-09Gravatar A. Unique TensorFlower2018-10-09
| | | | PiperOrigin-RevId: 216323343
* Enable support for PRED values in KeyValueSort for the HloEvaluator.Gravatar Adrian Kuegel2018-10-09
| | | | PiperOrigin-RevId: 216315110
* Automated rollback of commit 5f308cb408eb46ec9af0546be6b9ae1d5166b185Gravatar A. Unique TensorFlower2018-10-08
| | | | PiperOrigin-RevId: 216309111
* Refactor CalculateOutputSize() from VirtualScheduler protected member ↵Gravatar Peter Ma2018-10-08
| | | | | | function to utils; Refactor EstimateSize() from memory_optimizer.cc to utils; some small changes for readability improvement PiperOrigin-RevId: 216307257
* Add Floor_mod to schema.Gravatar A. Unique TensorFlower2018-10-08
| | | | PiperOrigin-RevId: 216303340
* Automated rollback of commit 07df147ab20c4a5329148e5fb5f7f6b187cb73a4Gravatar Reed Wanderman-Milne2018-10-08
| | | | PiperOrigin-RevId: 216299809
* [XLA] Introduce input/output alias config.Gravatar Yunxing Dai2018-10-08
| | | | | | | | - This CL intruduces input/output alias config in HLO module that allows any HLO pass to configure it. Once the alias_config is set, each backend needs to follow the contract during execution time to make sure the input and output are indeed aliased. - Copy insertion / buffer assignment and alias analysis has been updated to correctly honor the config and avoid any possible liveness interference. PiperOrigin-RevId: 216299501
* Add a tracing::ScopedActivity event to track the duration of a Session::Run()Gravatar A. Unique TensorFlower2018-10-08
| | | | | | | call for better xprof tracing. Also annotate synchronous op execution with the session-run id (or step_id) as metadata leveraging the support introduced in cl/215985561. This should enable highlighting the duration of a Session::Run and all the ops that ran in it for visualizing latency regressions in the case of CPU inference. PiperOrigin-RevId: 216284682
* Register int64 SUM GPU kernel.Gravatar James Qin2018-10-08
| | | | PiperOrigin-RevId: 216280913
* Fix the seeding for `Dataset.shuffle(..., reshuffle_each_iteration=False)`.Gravatar Derek Murray2018-10-08
| | | | | | | | | Previously, we were passing the first (graph-level) seed for both the graph-level and op-level seeds when creating a C++ dataset. This change passes the op-level seed to the appropriate point, and adds a test for the behavior with graph-but-not-op-level seeds. PiperOrigin-RevId: 216280641
* Consolidate device parameter arguments into a shared DeviceInfo structGravatar A. Unique TensorFlower2018-10-08
| | | | PiperOrigin-RevId: 216280197
* Remove deprecations for some of the endpoints in ApiDef files. These changesGravatar Anna R2018-10-08
| | | | | | | | | | | | | | | are made according to https://github.com/tensorflow/community/pull/16. I am keeping a few symbols deprecated not mentioned in the doc: tf.diag - it seems best to keep it next to tf.linalg.diag, so that the two are easy to compare and decide which one to use. The plan is to rename tf.diag to tf.tensor_diag. tf.is_nan - similar to tf.is_inf, tf.is_finite, tf.is_numeric_tensor which are all getting deprecated and replaced by symbols in tf.debugging. tf.string_to_number - other string endpoints in root namespace are getting deprecated: for e.g. tf.substr, tf.string_join. tf.dequantize - all quantization ops should be under tf.quantize. I probably missed this one. tf.check_numerics - similar to other debugging ops that are getting moved to tf.debugging. tf.squared_difference - moved to tf.math namespace and not as popular as some other math ops such as tf.add to justify keeping endpoint in root. tf.decode_raw - similar to other ops such as tf.decode_csv that are getting moved to tf.io.decode_csv. PiperOrigin-RevId: 216278010
* Avoid calling get_default_graph() during tf.enable_eager_execution()Gravatar Gunhan Gulsoy2018-10-08
| | | | PiperOrigin-RevId: 216270497
* Internal changeGravatar Jared Duke2018-10-08
| | | | PiperOrigin-RevId: 216270385
* Convert TensorFlow's aws dependency to new third party import method.Gravatar A. Unique TensorFlower2018-10-08
| | | | PiperOrigin-RevId: 216265275
* [XLA] Make overly-specific ShapeUtil predicate a little more general.Gravatar Chris Leary2018-10-08
| | | | PiperOrigin-RevId: 216263039
* Automated rollback of commit 13b47e6c4f9d7b295948b1057139bf676e394b6fGravatar Derek Murray2018-10-08
| | | | PiperOrigin-RevId: 216260575
* Internal Change.Gravatar Michael Case2018-10-08
| | | | PiperOrigin-RevId: 216260437
* Simple comment fix in CheckpointInputPipelineHook.Gravatar Ruoxin Sang2018-10-08
| | | | PiperOrigin-RevId: 216260216
* Fix issue with type inference for ops with fixed output typesGravatar Jared Duke2018-10-08
| | | | | | | Use the ArgDef::type field when available for propagating the output types from a given unsupported operator. PiperOrigin-RevId: 216257741
* Automated rollback of commit 09b0fc199129e0f487a39741bdf674cf09035cbcGravatar Derek Murray2018-10-08
| | | | PiperOrigin-RevId: 216256115
* [XLA] Simplify loop nesting in HandleConvolutionGravatar David Majnemer2018-10-08
| | | | | | | | | | | | | | | The calculation of a spatial coordinate in the kernel and activations is not dependent on which part of the contracted dimension (input feature) we are in. Rather than nesting the loops, the loops can be siblings: - One loop over spatial dimensions - One loop over the input feature group This reduces the nesting depth which makes the code a little more readable and might be slightly faster due work invariant in the spatial loop getting hoisted out. PiperOrigin-RevId: 216255839
* Ignore args and kwargs for defun's get_concrete_fn if `PolymorphicFunction` ↵Gravatar Shivani Agrawal2018-10-08
| | | | | | | | was created with an input_signature. PiperOrigin-RevId: 216253122
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-25 23:04:38 +0000