experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/mlir/xla
History
Smit Hinsu a309b4ee31 Fix TensorFlow ScatterNd op lowering to HLO 
* Use TensorScatterAdd to correctly handle repeated indices
* Support Complex typed operands

Also, enabled complex typed operands for ZerosLikeOp.

PiperOrigin-RevId: 338347731
Change-Id: Ieade5166c3d8e234bda2f090bc636dc6c98931b1
2020-10-21 15:47:51 -07:00
..
ir
Make mhlo.sort return variadic results instead of a tuple
2020-09-28 13:47:25 -07:00
tests
Export HLO sort ops with a single operand to XLA HLO
2020-10-21 15:41:20 -07:00
transforms
Fix TensorFlow ScatterNd op lowering to HLO
2020-10-21 15:47:51 -07:00
attribute_importer.cc
Rename xla_hlo dialect to mhlo
2020-07-06 21:54:22 -07:00
attribute_importer.h
Rename xla_hlo dialect to mhlo
2020-07-06 21:54:22 -07:00
BUILD
Scatter HLO -> LHLO conversion test
2020-10-21 14:36:46 -07:00
hlo_function_importer.cc
[mlir] Add FusionOp to XLA HLO -> LMHLO
2020-10-13 14:20:52 -07:00
hlo_function_importer.h
[mlir] Add FusionOp to XLA HLO -> LMHLO
2020-10-13 14:20:52 -07:00
hlo_module_importer.cc
Explicitly load standard dialect in HLO importer (NFC)
2020-08-23 10:51:33 -07:00
hlo_module_importer.h
Explicitly load Mhlo dialect in HLO importer (NFC)
2020-08-22 21:49:11 -07:00
hlo_to_mlir_hlo.cc
Initial version of HLO translation tools.
2019-07-17 19:59:01 -07:00
hlo_to_mlir_hlo.h
Rename xla_hlo dialect to mhlo
2020-07-06 21:54:22 -07:00
hlo_utils.cc
Move bfloat16 header to tensorflow/core/platform.
2020-08-14 13:44:57 -07:00
hlo_utils.h
Rename xla_hlo dialect to mhlo
2020-07-06 21:54:22 -07:00
mlir_hlo_to_hlo.cc
Export HLO sort ops with a single operand to XLA HLO
2020-10-21 15:41:20 -07:00
mlir_hlo_to_hlo.h
[MLIR] Add a switch for enabling layouts in MLIR -> XLA HLO
2020-10-02 13:42:55 -07:00
operator_writer_gen.cc
Add initial support for populating XLA OpMetadata with operation location when exporting MLIR HLO to HLO proto.
2020-08-31 18:32:51 -07:00
type_to_shape_test.cc
[MLIR/XLA] Fix two bugs:
2020-10-13 14:49:17 -07:00
type_to_shape.cc
[MLIR/XLA] Fix two bugs:
2020-10-13 14:49:17 -07:00
type_to_shape.h
Change comment of the external repo #include file origin.
2020-03-23 10:42:53 -07:00
xla_mlir_translate_cl.cc
[mlir] Add FusionOp to XLA HLO -> LMHLO
2020-10-13 14:20:52 -07:00
xla_mlir_translate_cl.h
[mlir] Add FusionOp to XLA HLO -> LMHLO
2020-10-13 14:20:52 -07:00
xla_mlir_translate.cc
[MLIR] Add a switch for enabling layouts in MLIR -> XLA HLO
2020-10-02 13:42:55 -07:00
xla_mlir_translate.h
mlir-translate: support -verify-diagnostics
2019-11-07 11:47:37 -08:00