experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use mlir::OpState::operator->() to get to methods of mlir::Operation.

Browse Source 
This is a preparation step to remove those methods from OpState.

PiperOrigin-RevId: 348436381
Change-Id: I814d25b2bdab58b7123ff33b7793c7c3a4fcf7cf
This commit is contained in:
Christian Sigg 2020-12-21 02:29:14 -08:00 committed by TensorFlower Gardener
parent 5ce6ac8cb9
commit 7d034e94dc
1 changed files with 2 additions and 2 deletions

4
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
View File

@ -69,8 +69,8 @@ inline LogicalResult GetIntegerArraySafe(
LogicalResult ConvertMaxUnpoolingFunc::RewriteFunc() {
func_.eraseBody();
func_.addEntryBlock();
func_.setAttr(kTFImplements,
StringAttr::get(kMaxUnpooling, func_.getContext()));
func_->setAttr(kTFImplements,
StringAttr::get(kMaxUnpooling, func_.getContext()));
OpBuilder builder(func_.getBody());
std::string custom_option_buffer;