STT-tensorflow/tensorflow/compiler/mlir/xla
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
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