Searched refs:GetNumReplicatedBlockArguments (Results 1 – 3 of 3) sorted by relevance
607 unsigned ReplicateOp::GetNumReplicatedBlockArguments() { in GetNumReplicatedBlockArguments() function in mlir::tf_device::ReplicateOp626 return block_arg.getArgNumber() < GetNumReplicatedBlockArguments(); in IsReplicatedBlockArgument()666 unsigned num_replicated_args = GetNumReplicatedBlockArguments(); in GetOperandsForBlockArgument()
286 unsigned GetNumReplicatedBlockArguments();
242 replicated_inputs.reserve(replicate.GetNumReplicatedBlockArguments()); in AddInputsToReplicateOp()409 replicate.GetNumReplicatedBlockArguments() - 1)); in HandleReplicateOp()