Searched defs:cluster_to_core_index (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | xla_sharding_util.cc | 370 llvm::SmallVector<llvm::SmallVector<int, 4>, 4> cluster_to_core_index, in LookupClusterToCoreIndex() argument
392 llvm::SmallVector<llvm::SmallVector<int, 4>, 4> cluster_to_core_index, in GetTileShardedOutputsToMerge() argument
419 llvm::SmallVector<llvm::SmallVector<int, 4>, 4> cluster_to_core_index, in HandleTileShardedOutputs() argument
515 llvm::SmallVectorImpl<int>* cluster_to_core_index) { in GetOutputTypesForLogicalDeviceComputation()
556 llvm::SmallVector<llvm::SmallVector<int, 4>, 4> cluster_to_core_index, in RemapOutputsFromLogicalDevices() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_rewrite_pass.cc | 444 llvm::SmallVector<int, 4> cluster_to_core_index; in BuildExecuteOp() local
509 llvm::SmallVectorImpl<llvm::SmallVector<int, 4>>* cluster_to_core_index, in AddToParallelExecuteOp()
829 llvm::SmallVector<llvm::SmallVector<int, 4>, 4> cluster_to_core_index; in Rewrite() local
|