- 31 10月, 2020 40 次提交
-
-
由 Meghna Natraj 提交于
PiperOrigin-RevId: 339959753 Change-Id: I492d1f750e7a6fd97d4fdccbb838492e699c56b6
-
由 Anna R 提交于
PiperOrigin-RevId: 339956677 Change-Id: Id4cea5f63b3f257170c121d74e9cb2995a10b629
-
由 Meghna Natraj 提交于
PiperOrigin-RevId: 339954904 Change-Id: Id9f6717da5c32bff185a10a37e9682be64cc6501
-
由 Nat Jeffries 提交于
PiperOrigin-RevId: 339953940 Change-Id: I568589fc09fa6e5aae137e7e8533e1dab86190d3
-
由 Meghna Natraj 提交于
PiperOrigin-RevId: 339951415 Change-Id: Ic3bbb2e474a4fb287fb48f24509beb44625ee2e4
-
由 Anna R 提交于
PiperOrigin-RevId: 339951343 Change-Id: I02b02143721f7c656af8c1d19176dc44f3bffa20
-
由 Scott Zhu 提交于
PiperOrigin-RevId: 339948109 Change-Id: Ia2612efa66b48ac5784d2a53cdeb0ce9001f4025
-
由 TensorFlower Gardener 提交于
PiperOrigin-RevId: 339946530 Change-Id: I9ff4d4a380b6f2c2181edfc76cbb41af09abf0e2
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 339944681 Change-Id: Ifd0f43a2616328fe54819fae18ee626a44556e99
-
由 Ruoxin Sang 提交于
Dataset cardinality op should be placed on the same host device as the dataset, otherwise TF will try to encode/decode the dataset to another host device, which is not possible. PiperOrigin-RevId: 339942622 Change-Id: Ifbeb0aabdfc35f6c0e1248d69a8d22727e19c4a2
-
由 Yunxing Dai 提交于
This further simplifies ShapedBuffer object as itself doesn't have any logic to use the platform field. Notice that this cl also removed some sanity check in allocation tracker. we can add that sanity check back if need --- just keep track of `platform` inside of allocation tracker as a side map. PiperOrigin-RevId: 339938197 Change-Id: I090e603927ed3fccdb51254f972b3af2e1ec1470
-
由 TensorFlower Gardener 提交于
PiperOrigin-RevId: 339937575 Change-Id: I1d80b244adcb27a429bc09c63434847107d23bee
-
由 Rick Chao 提交于
PiperOrigin-RevId: 339929158 Change-Id: I8592aa6e2cec32a2ba97743a6f022f263f0f65e2
-
由 Hongmin Fan 提交于
TFRT batch fallback kernel to crash in high QPS load. The bug makes it only create an object of base class BatchTask even when splitting a task of the derived class FallbackBatchTask (used only in TFRT), and put it into a batch with other FallbackBatchTask objects. When this batch of mixed types of tasks is processed, it crashes. PiperOrigin-RevId: 339927837 Change-Id: Ie52bd11c61c9ddbe6ab803cd90208419d4b2dba6
-
由 Anna R 提交于
PiperOrigin-RevId: 339922525 Change-Id: I88c0d616154484ddd353d262f2fe7408cfc058c4
-
由 Katherine Wu 提交于
#KERAS_PRIVATE_API_CLEANUP PiperOrigin-RevId: 339917969 Change-Id: Iadd13a8d23a941528e0384e090dc07ddc9d63da6
-
由 Thomas O'Malley 提交于
With this change, it is now possible to mix-and-match tf.keras.Layers and tf.Modules inside a tf.keras.Model and everything will be tracked properly. - Variables in tf.Modules that are set as attributes of custom Layers and Models now show up properly in properties such as Layer.trainable_variables and Model.trainable_variables. - tf.Modules do not show up in Model.layers. Instead, a new method Layer._flatten_modules is added that iterates over tf.Modules and Layers in the order that Keras expects. The existing method Layer.submodules (inherited from tf.Module) can still be used to iterate over tf.Modules and Layer with the tf.Module ordering. Layer._flatten_layers is built on top of Layer._flatten_modules - Layer._layers is renamed to Layer._self_tracked_trackables to avoid naming conflicts with user-defined attributes (and to reflect that this attr contains Layers, Modules, and TrackableDataStructures) - A new property is added to tf.Module to enable this, namely tf.Module.non_trainable_variables PiperOrigin-RevId: 339917644 Change-Id: I96a7302745280a6261de8c4295c5cbf5f4d7dd5c
-
由 Jiho Choi 提交于
PiperOrigin-RevId: 339915378 Change-Id: I5675fdf910a3d8fd550d28a5382f0da20e450281
-
由 Raman Sarokin 提交于
PiperOrigin-RevId: 339913277 Change-Id: If4a92b3fafe922b5abf00bc8dc04deef1ef6ca6d
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 339911119 Change-Id: Iddf765ff3e58ea67c0cdf3b1ffe3065a92e8dcfc
-
由 Advait Jain 提交于
This change avoids an implicit conversion from enum BuiltinOperator to int8_t for calls to CreateOperatorCodeDirect. This should mean that once the BuiltinOperator enum is larger than a byte, the TFLM code does not get tripped. PiperOrigin-RevId: 339910848 Change-Id: Ief978f2bb9ea818f8211ade3927baa35dc7a8512
-
由 Katherine Wu 提交于
Add property that allows layers to specify that the input_spec can also be used as the layer call function's input_signature. By default,all Keras exported layers will have this property return True, since Keras more rigidly defines the input_spec shape (compared to user-defined models, which may only have `ndims` set). PiperOrigin-RevId: 339909813 Change-Id: I1b486747aa1e413e6f24f6809fb88846ad4712ab
-
由 Scott Zhu 提交于
PiperOrigin-RevId: 339909803 Change-Id: I7cbcc1d74c0de2cd7f86e10e2b44f0d4b81274a0
-
由 Jose Baiocchi 提交于
PiperOrigin-RevId: 339909494 Change-Id: I1a9352342e9b79b55b5ba1f324fecf16a34b7647
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 339906858 Change-Id: I0ad0e7fef95d634a75ddb22a5b08dee9b66c2ec1
-
由 Jiri Simsa 提交于
PiperOrigin-RevId: 339906797 Change-Id: Ieff06a2d26a3b7bfc90987bba8457f1ff866f8af
-
由 TensorFlower Gardener 提交于
PiperOrigin-RevId: 339899776 Change-Id: Iff5a37227f189607727f3e24d75b41ee8b906af9
-
由 TensorFlower Gardener 提交于
PiperOrigin-RevId: 339899425 Change-Id: If9bbb9c68787dc484129102c716da2203c209dab
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 339893950 Change-Id: I6d3bf00d3296f5008ce2bd9f208dc0600a515665
-
由 Jian Li 提交于
PiperOrigin-RevId: 339893949 Change-Id: I43c4d0c58a19efb52b66a2ccbf9f792f05e0c73a
-
由 Thomas O'Malley 提交于
properties to help distinguish between metrics that should and shouldn't be passed sample_weight argument. Note these properties are set to None before Model.fit is called, since metrics are potentially broadcast to match the structure of data seen in Model.fit. PiperOrigin-RevId: 339892649 Change-Id: I0abffae08efde2b8adc58014ef205d318d66a9ab
-
由 Tomer Kaftan 提交于
PiperOrigin-RevId: 339892095 Change-Id: Ic6c9cebe68aad3d2b388bcdfa3e7249628dcf8ac
-
由 Tres Popp 提交于
PiperOrigin-RevId: 339890070 Change-Id: I8144220b5ffc20132e1bb2fd0b099266d2a0391c
-
由 Stephan Herhut 提交于
This adds back functionality to the unranked case that was already present (albeit manual) in the ranked case. PiperOrigin-RevId: 339889861 Change-Id: I6426ac5553c75bebd829135104d1b7802ff2a254
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 339888534 Change-Id: I60fb13fad42ec5938d7276151afdce2dc3a6926f
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 339887960 Change-Id: I536c60018b983c115c27d664b57fb268ea90ee98
-
由 Kuangyuan Chen 提交于
PiperOrigin-RevId: 339887058 Change-Id: I2decd73c27e6c3dc1a79cc723075598025827852
-
由 Marat Dukhan 提交于
Pull a work-around for compilation of x86 versions with legacy compilers PiperOrigin-RevId: 339886664 Change-Id: I5035f38cff401564061f092f59d5dea91313f5bf
-
由 Stephan Herhut 提交于
This will replace the existing, hardwired pass that was used for ranked kernels. PiperOrigin-RevId: 339886355 Change-Id: I362f04a806a0079733caa91edff73d2547fca221
-
由 A. Unique TensorFlower 提交于
PiperOrigin-RevId: 339883324 Change-Id: I9b8cbe92bfa2bd279be4d72ca9a40b894f7fcaa7
-