Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dtrivial_spmd_expander.cc85 for (const auto& result_and_index : llvm::enumerate(op->getOpResults())) { in ComputeLayoutForward() local
86 const int index = result_and_index.index(); in ComputeLayoutForward()
87 auto result = result_and_index.value(); in ComputeLayoutForward()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc519 for (auto result_and_index : llvm::enumerate(cluster_func.getResults())) { in GetOutputTypesForLogicalDeviceComputation() local
520 const auto output_index = result_and_index.index(); in GetOutputTypesForLogicalDeviceComputation()
524 result_and_index.value().getType().cast<mlir::TensorType>(); in GetOutputTypesForLogicalDeviceComputation()
560 for (auto& result_and_index : in RemapOutputsFromLogicalDevices() local
562 const auto output_index = result_and_index.index(); in RemapOutputsFromLogicalDevices()
563 const auto old_parallel_execute_output = result_and_index.value(); in RemapOutputsFromLogicalDevices()