mlir-hlo

Commit Graph

Author	SHA1	Message	Date
Robert Suderman	06ae59074f	Fold xla iota across a 1-length dimension into a zero value Iota across length-1 is just a constant. Fold into it. PiperOrigin-RevId: 320443468	2020-07-30 22:33:43 +00:00
Uday Bondhugula	de0578b4f9	PR #40745 : [MLIR] Add constant folder for xla_hlo.broadcast_in_dim op Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/40745 Fold broadcast_in_dim op if the operand is the result of a tensor splat. Copybara import of the project: -- 26c9f631448b8d6ffd20ece39ea8d4132b5550c7 by Uday Bondhugula <uday@polymagelabs.com>: [MLIR] Add constant folder for xla_hlo.broadcast_in_dim op Fold broadcast_in_dim op if the operand is the result of a tensor splat. COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/tensorflow/pull/40745 from polymage-labs:broadcast_in_dim_fold 26c9f631448b8d6ffd20ece39ea8d4132b5550c7 PiperOrigin-RevId: 320365164	2020-07-30 22:33:34 +00:00
Robert Suderman	e1651b6090	Canonicalize multidimensional iota to use broadcast There is no reason to have a multidimensional iota for codegen. This should be canonicalized to a single dimensional iota followed by a broadcast. Changing iota to on a single dimension and a broadcast substantially simplifies implementing iota operations. PiperOrigin-RevId: 320095470	2020-07-30 22:32:54 +00:00
Mehdi Amini	8900222fed	Rename `xla_hlo` dialect to `mhlo` This is part of the current refactoring of the HLO related dialect. `xla_hlo` will be reintroduced in a new form later. PiperOrigin-RevId: 319916753	2020-07-30 22:32:50 +00:00
Mehdi Amini	fa057cc0bc	Move HLO tests to `third_party/tensorflow/compiler/mlir/hlo/` Also add a localized `mlir-hlo-opt` binary for the testing of tensorflow/compiler/mlir/hlo/... ; this directory is intended to be self-contained and depend only on MLIR. PiperOrigin-RevId: 319878984	2020-07-30 22:32:45 +00:00

5 Commits