| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 212668653
|
|
|
|
|
|
|
|
| |
This now matches the definition. I fixed it here rather than in the definition as it seems every call to this function names the variable "num_components".
I also tidied up the comment a little.
PiperOrigin-RevId: 212668416
|
|
|
|
|
|
| |
tests
PiperOrigin-RevId: 212668288
|
|
|
|
|
|
| |
Just for completeness, this isn't being used currently.
PiperOrigin-RevId: 212667528
|
|\
| |
| |
| | |
PiperOrigin-RevId: 212667200
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212658047
|
| |
| |
| |
| |
| |
| | |
optimization pass, instead of a step in XlaCompiler.".
PiperOrigin-RevId: 212657932
|
| |
| |
| |
| |
| |
| |
| | |
will remove this from contrib.
linear_operator_addition is hidden from the public API.
PiperOrigin-RevId: 212655087
|
| |
| |
| |
| |
| |
| |
| |
| | |
copying that type.
This avoids unnecessary string copies and deallocations in the ReadVariableOp, and similar ops.
PiperOrigin-RevId: 212652588
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212651704
|
| |
| |
| |
| |
| |
| | |
The test would fall back to GPU:0 when unable to find a GPU. This should be CPU.
PiperOrigin-RevId: 212649435
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| | |
TFLite has an optimized matmul path for cases where the input and
filter tensors have matching width+height. However, this case doesn't
properly account for multiple *batches*. Account for this and add
an appropriate test.
Credit to zgxnet for the bug and proposed fix.
Fixes #21817
PiperOrigin-RevId: 212645329
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212645190
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212643986
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212642629
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212601523
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212600364
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212595533
|
| |
| |
| |
| |
| |
| | |
correctly. Refactor MapDefunOp for correctness + better C++ hygiene
PiperOrigin-RevId: 212589285
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212582822
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212577288
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212569958
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212568029
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212565231
|
| |
| |
| |
| |
| |
| | |
performance.
PiperOrigin-RevId: 212557406
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212556651
|
| |
| |
| |
| |
| |
| | |
it more accurate.
PiperOrigin-RevId: 212555968
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212555263
|
| |
| |
| |
| | |
PiperOrigin-RevId: 212551965
|
|\ \
| | |
| | |
| | | |
PiperOrigin-RevId: 212551803
|
| | |
| | |
| | |
| | | |
PiperOrigin-RevId: 212548263
|
| | |
| | |
| | |
| | | |
PiperOrigin-RevId: 212545735
|
| | |
| | |
| | |
| | |
| | |
| | | |
the frontend autograph module in contrib for backward compatibility. Certain files, like notebooks and the readme file may be referenced from the outside, so a copy of those is kept as well. In addition, the notebooks subdirectory of examples is also kept in contrib because the extension the build file relies on is not available in the PIP package.
PiperOrigin-RevId: 212543067
|
|\ \ \
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212543062
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212542938
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
any pip builds.
PiperOrigin-RevId: 212541571
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
02451914b9ad5320f81f56a89f3eef1f8683227c
PiperOrigin-RevId: 212539233
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212535619
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212535448
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212534671
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212533869
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
Allow each Resource to manage multiple streams that share the same quantile config -- number of quantiles and epsilon. Previously each resource manage only one stream, so we will have to create resources equal to the number of features, which is cumbersome when input is high dimensional. If 1000 features use 100 quantiles (which is hardcoded today), then 1000 resources is required. This cl will create the number of resources linear to the number of parameter servers, if 2 parameter servers are present, then only 2 resources is required, one for each ps.
Remove time stamp token as the ops are called once.
PiperOrigin-RevId: 212533735
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
to see the total cost of running Grappler in logs. Also add a couple of VLOG(1) statements to see breakdown between main graph and function optimization.
PiperOrigin-RevId: 212531430
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
input_layer.
PiperOrigin-RevId: 212528172
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212524769
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212521825
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212518982
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212517784
|
| | | |
| | | |
| | | |
| | | | |
PiperOrigin-RevId: 212516939
|
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | |
| | | | |
Instead of multiple primitive Eigen ops in Conv3DBackpropInput, call directly into the ex-NeuralNetworks module's function CuboidConvolutionBackwardInput.
Modest ~10% latency improvement and ~15-20% peak memory reduction.
PiperOrigin-RevId: 212516586
|