Home
last modified time | relevance | path

Searched refs:output_sharding_attrs (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc317 const auto output_sharding_attrs = in ParseAndValidateOutputSharding() local
320 if (!output_sharding_attrs) in ParseAndValidateOutputSharding()
324 if (output_sharding_attrs.size() != cluster_func.getNumResults()) in ParseAndValidateOutputSharding()
328 llvm::enumerate(output_sharding_attrs)) { in ParseAndValidateOutputSharding()