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: 348838982
Change-Id: Iba4c3b5208a0b8352b06536660ba141b1574c445
This commit is contained in:
Christian Sigg 2020-12-23 13:20:27 -08:00 committed by TensorFlower Gardener
parent 60b9283c9e
commit 208bf5695b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/mlir/tensorflow/transforms/tpu_sharding_identification_pass.cc
View File

@ -306,8 +306,8 @@ void IdentifyXlaShardingForTPUComputation(
xla::sharding_builder::AssignDevice(0).SerializeAsString(); xla::sharding_builder::AssignDevice(0).SerializeAsString();
bool use_spmd = false; bool use_spmd = false;
if (auto use_spmd_attr = if (auto use_spmd_attr = cluster_func->getAttrOfType<BoolAttr>(
cluster_func.getAttrOfType<BoolAttr>("use_spmd_for_xla_partitioning")) "use_spmd_for_xla_partitioning"))
use_spmd = use_spmd_attr.getValue(); use_spmd = use_spmd_attr.getValue();
IdentifyXlaShardingForComputationInputs(logical_core_0_sharding, use_spmd, IdentifyXlaShardingForComputationInputs(logical_core_0_sharding, use_spmd,