Thai Nguyen
e9781e9b16
Support TFLite in the tool to print selective registration header
...
PiperOrigin-RevId: 314282031
Change-Id: Ie71b434c177d03e246a5cfde3d067ac695b71299
2020-06-02 00:49:20 -07:00
A. Unique TensorFlower
a78fa541d8
Replace list comprehension with generator expressions.
...
PiperOrigin-RevId: 285822581
Change-Id: I679256cc6f5890fa93ff3a2bfb9136b5d679d3ac
2019-12-16 12:26:12 -08:00
Amit Patankar
7fa4cfb49d
Export the kernel_registry functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information.
...
PiperOrigin-RevId: 275858306
Change-Id: I816f1f43f765a1e499b8f5d8fe168e0c01f9dab5
2019-10-21 09:31:12 -07:00
A. Unique TensorFlower
75390d4c35
Special-case the AccumulateNV2 op in print_selective_registration_header
...
AccumulateNV2 doesn't have or need a kernel. It gets rewritten to other ops by
accumulate_n_optimizer.cc. This change allows it to be mentioned in the output
of print_selective_registration_header, rather than being ignored with a
warning. Behavior for other ops is preserved.
PiperOrigin-RevId: 211701878
2018-09-05 14:47:01 -07:00
Vivek Rane
d34b523f8f
Fixed 3 bugs picked up by the unit tests ( #16055 )
2018-01-12 12:52:13 -08:00
A. Unique TensorFlower
f333979e0d
Change the header generated for selective registration to use the isequals
...
defined in the header, instead of strcmp. strcmp is not a constexpr on all
platforms.
PiperOrigin-RevId: 156914421
2017-05-23 14:52:33 -07:00
A. Unique TensorFlower
a311a55c46
Print an "autogenerated" message into selective registration headers.
...
Change: 151759343
2017-03-30 15:48:04 -07:00
Manjunath Kudlur
0e7edf20ee
Make selective registration handle spaces in kernel name.
...
Change: 148837641
2017-02-28 17:46:55 -08:00
A. Unique TensorFlower
a00adb1114
Split print_selective_registration_header into library and binary.
...
Change: 148358318
2017-02-23 10:41:53 -08:00
