Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc506 for (auto result_and_index : llvm::enumerate(cluster_func.getResults())) { in GetOutputTypesForLogicalDeviceComputation() local
507 const auto output_index = result_and_index.index(); in GetOutputTypesForLogicalDeviceComputation()
511 result_and_index.value().getType().cast<mlir::TensorType>(); in GetOutputTypesForLogicalDeviceComputation()
542 for (auto result_and_index : llvm::enumerate(cluster_func.getResults())) { in RemapOutputsFromLogicalDevices() local
543 const auto output_index = result_and_index.index(); in RemapOutputsFromLogicalDevices()
544 const auto cluster_func_output = result_and_index.value(); in RemapOutputsFromLogicalDevices()