Searched refs:GetOperandsForBlockArgument (Results 1 – 4 of 4) sorted by relevance
639 MutableArrayRef<OpOperand> operands = GetOperandsForBlockArgument(block_arg); in GetReplicaOperandIndexForBlockArgument()650 MutableArrayRef<OpOperand> operands = GetOperandsForBlockArgument(block_arg); in GetReplicaOperandForBlockArgument()661 MutableArrayRef<OpOperand> ReplicateOp::GetOperandsForBlockArgument( in GetOperandsForBlockArgument() function in mlir::tf_device::ReplicateOp
294 MutableArrayRef<OpOperand> GetOperandsForBlockArgument(BlockArgument block_arg);
174 replicate.GetOperandsForBlockArgument(replicate_arg); in HandleReplicatedInputs()
415 replicate.GetOperandsForBlockArgument(block_arg); in HandleHostReplicatedInputs()