Searched refs:operand_and_idx (Results 1 – 3 of 3) sorted by relevance
323 for (auto operand_and_idx : llvm::enumerate(fetch.getOperands())) { in AddEdge() local324 Value operand = operand_and_idx.value(); in AddEdge()327 auto* dst_node = returns_[fetch][operand_and_idx.index()]; in AddEdge()361 for (auto operand_and_idx : llvm::enumerate(op->getOperands())) in AddEdge() local362 TF_RETURN_IF_ERROR(AddEdgeBetweenNodes(operand_and_idx.value(), dst_node, in AddEdge()363 operand_and_idx.index())); in AddEdge()369 for (auto operand_and_idx : llvm::enumerate(inst->getOperands())) in AddEdge() local371 AddEdgeBetweenNodes(operand_and_idx.value(), dst_node, in AddEdge()372 operand_and_idx.index() + operand_offset)); in AddEdge()418 for (auto operand_and_idx : llvm::enumerate(fetch.getOperands())) { in AddFetchNode() local[all …]
64 for (auto operand_and_idx : llvm::enumerate(cluster_func.getOperands())) { in GetRemappedReplicatedInputIndices() local65 if (auto block_arg = operand_and_idx.value().dyn_cast<BlockArgument>()) { in GetRemappedReplicatedInputIndices()72 remapped_indices[replicated_input_index] = operand_and_idx.index(); in GetRemappedReplicatedInputIndices()
374 for (auto operand_and_idx : llvm::enumerate(cluster_func.getOperands())) { in BuildCompileOp() local377 metadata.args(operand_and_idx.index()).shape()); in BuildCompileOp()383 operand_and_idx.value()); in BuildCompileOp()