- 23 2月, 2019 17 次提交
-
-
由 TensorFlower Gardener 提交于
PiperOrigin-RevId: 235218660
-
由 Derek Murray 提交于
Previously, the tuple would be unpacked in string formatting, and if it had more than one element we would get an argument unpacking error, instead of the intended TypeError. PiperOrigin-RevId: 235217943
-
由 Derek Murray 提交于
Internal users of these properties should now use dataset_ops.get_legacy_output_{shapes,types,classes}() instead. In a future CL we will remove the properties from the DatasetV2. PiperOrigin-RevId: 235217553
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235215099
-
由 TensorFlower Gardener 提交于
PiperOrigin-RevId: 235214785
-
由 A. Unique TensorFlower 提交于
Provides better memory locality which speeds up most ops, and makes destruction trivial. PiperOrigin-RevId: 235213734
-
由 Allen Lavoie 提交于
An unfortunate combination of the name_scope method wrapping not forwarding argspecs and input_signatures inspecting functions for keyword arguments. PiperOrigin-RevId: 235211265
-
由 A. Unique TensorFlower 提交于
Notice that most of the constants created by the constant folding pass are already in compressed format. This just extends this practice to all constants in the graph that have at least 64 elements and where the achievable compression ratio is grater or equal to 2. PiperOrigin-RevId: 235211073
-
由 Bruce Fontaine 提交于
PiperOrigin-RevId: 235210894
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235210508
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235207900
-
由 Zhenyu Tan 提交于
PiperOrigin-RevId: 235207665
-
由 Rohan Jain 提交于
Renames to prepare for the TF2.0 API. HashTable -> StaticHashTable and MutableDenseHashTable -> DenseHashTable. Also insert -> insert_or_assign and remove -> erase for DenseHashTable. PiperOrigin-RevId: 235206233
-
由 Guangda Lai 提交于
PiperOrigin-RevId: 235201217
-
由 Dan Moldovan 提交于
PiperOrigin-RevId: 235197577
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235197341
-
由 Dan Moldovan 提交于
PiperOrigin-RevId: 235192783
-
- 22 2月, 2019 23 次提交
-
-
由 Brian Lee 提交于
PiperOrigin-RevId: 235184105
-
由 James Keeling 提交于
This function previously stated that it could accept a rank-0 Tensor as its `num_or_size_splits` argument. This is in fact not correct: TF graphs must have a static number of output tensors for any operation, so the number of splits has to be known statically or we cannot validate the correctness of the graph. This method will now raise a ValueError if called with a rank-0 Tensor. I also make the documentation more explicit. Strangely, the test was testing for the incorrect behaviour, so I also fixed that. PiperOrigin-RevId: 235157900
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235149900
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235144234
-
由 Blake Hechtman 提交于
PiperOrigin-RevId: 235138912
-
由 Nick Felt 提交于
Previously, when in a GPU device context, the logic attempted to create the default empty metadata tensor on the GPU, which fails since it's a string tensor. PiperOrigin-RevId: 235135723
-
由 A. Unique TensorFlower 提交于
Add implementations for CUDA library functions which dynamically load the CUDA library DSO and forward the call to the actual function. Most of the code is auto-generated. Libraries implemented: CUDA driver API, cuRAND, cuFFT, cuBLAS, and cuDNN. The CUDA runtime is not implemented because it's not clear whether it should just be linked statically. Older versions of CUDA (starting with 8.0) and cuDNN (starting with 7.0) are handled with two approaches: - if the API didn't change and only new functions were added, declare types that were introduced afterwards. It is acceptable to implement APIs that were only introduced later: if they are called despite not being declared (in older headers), they will not be able to dlsym the corresponding function and return an error. - if the API changed, different contextual headers are included. PiperOrigin-RevId: 235135165
-
由 A. Unique TensorFlower 提交于
Replace angle bracket CUDA kernel launches ('<<<>>>') with API function calls ('cudaLaunchKernel'), through a helper function which also checks the return code in debug builds. PiperOrigin-RevId: 235135115
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235128104
-
由 Dan Moldovan 提交于
Handle the case when tf.function's target is a tf_decorator method that was hot-swapped after tf.function was applied. PiperOrigin-RevId: 235124593
-
由 A. Unique TensorFlower 提交于
Address TODO to support materialization of Fill nodes that are fed from constants stored in raw serialized format. PiperOrigin-RevId: 235123971
-
由 Saurabh Saxena 提交于
maximum_iterations no longer needs to be a TF-graph-compile-time constant but should be a compile-time constant when building the XLA graph (when values of placeholders are also available). There are broadly 2 parts of this change. 1. while_v2: This removes the _maximum_iterations attr from the functional While op generated by while_v2 and makes it an input instead. This frees maximum_iterations from being a graph-building time constant. Its index in the list of op inputs is preserved in both the forward and backwards graph so it should be ok for n'th order derivatives. 2. The XLA changes are needed to propagate constants inside the loop body e.g. when building the gradient of a While inside a while_loop. They are 2-fold: a. This updates the const_analysis pass to inspect the cond and body functions of While for compile time constant requirements. b. When compiling the XLA While op in while_op.cc if there are inputs that are compile time constants and the corresponding loop variables are loop invariants, the constants get propagated into the loop body. Makes XlaCompiler::FindFunctionBody public to allow accessing the body of the While in while_op.cc. If the body has been specialized using the PropagateConstIntoFunctionalNodes pass the rewritten function only exists in XlaCompiler::local_flib_def_. PiperOrigin-RevId: 235122516
-
由 Rohan Jain 提交于
it doesn't take a table as input and takes an initializer. Also oov_buckets should always be > 0 for this table. PiperOrigin-RevId: 235119219
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235117320
-
由 Yuefeng Zhou 提交于
PiperOrigin-RevId: 235114715
-
由 Stephan Lee 提交于
PiperOrigin-RevId: 235112772
-
由 A. Unique TensorFlower 提交于
resources will now be passed as 'data' instead of on the other attributes. END_PUBLIC For more information on the resources cleanup and rules, see: go/apple-rules:resources BEGIN_PUBLIC The "resources", "asset_catalogs", "bundles", "datamodels", "storyboards", "strings", and "xibs" attributes are directly moved to the "data" attribute. Any "structured_resources" are moved to a new apple_resource_group instance, and that is added to "data". PiperOrigin-RevId: 235112154
-
由 A. Unique TensorFlower 提交于
Change the Keras model_iteration() to correctly infer the size of validation datasets when validation_steps is None. PiperOrigin-RevId: 235112144
-
由 Jonathan Hseu 提交于
PiperOrigin-RevId: 235107020
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 235106933
-
由 David Majnemer 提交于
Pow(x, y) acts like std::pow which complicates trying to use Pow(x, 0.5) to implement sqrt. Pow(-0, 0.5) is +0 but Sqrt(-0) should be -0. As such, it makes sense to provide an explicit Sqrt and Rsqrt. PiperOrigin-RevId: 235105846
-
由 Smit Hinsu 提交于
PiperOrigin-RevId: 235104911
-
由 Stephan Lee 提交于
PiperOrigin-RevId: 235102987
-