tree 17390b25ce876d1077175af433606ab2c8176bd1
parent 6688e1c23b0f01400a02153408294e37295c5e96
author Alexander Belyaev <pifon@google.com> 1590673359 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1590673575 -0700

[XLA][MLIR] Register BufferAssignmentTestDialect.

It was made harder to shoot yourself in the foot when using operations that are
not registered with MLIRContext. In order to use test ops in the
buffer_assignment_test.mlir we have to register them first.

PiperOrigin-RevId: 313576961
Change-Id: Id3c711a2d1776a6fdee272d31d932ec5010cd0c2