Home
last modified time | relevance | path

Searched defs:output_shardings (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc276 auto output_shardings = in SetMetadataProtoRetvals() local
662 llvm::SmallVector<xla::OpSharding, 4> output_shardings; in Rewrite() local