Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc391 llvm::SmallVector<mlir::Value, 4> GetTileShardedOutputsToMerge( in GetTileShardedOutputsToMerge() function
421 auto outputs_to_merge = GetTileShardedOutputsToMerge( in HandleTileShardedOutputs()
583 auto tile_sharded_outputs = GetTileShardedOutputsToMerge( in RemapOutputsFromLogicalDevices()