Searched refs:GetOperandsForBlockArgument (Results 1 – 4 of 4) sorted by relevance
627 MutableArrayRef<OpOperand> operands = GetOperandsForBlockArgument(block_arg); in GetReplicaOperandIndexForBlockArgument()638 MutableArrayRef<OpOperand> operands = GetOperandsForBlockArgument(block_arg); in GetReplicaOperandForBlockArgument()649 MutableArrayRef<OpOperand> ReplicateOp::GetOperandsForBlockArgument( in GetOperandsForBlockArgument() function in mlir::tf_device::ReplicateOp
292 MutableArrayRef<OpOperand> GetOperandsForBlockArgument(BlockArgument block_arg);
191 replicate.GetOperandsForBlockArgument(replicate_arg); in HandleReplicatedInputs()
460 replicate.GetOperandsForBlockArgument(block_arg); in HandleHostReplicatedInputs()