Searched refs:GetReplicaOperandIndexForBlockArgument (Results 1 – 2 of 2) sorted by relevance
625 unsigned ReplicateOp::GetReplicaOperandIndexForBlockArgument( in GetReplicaOperandIndexForBlockArgument() function in mlir::tf_device::ReplicateOp
290 unsigned GetReplicaOperandIndexForBlockArgument(BlockArgument block_arg, unsigned replica);