Searched refs:GetNumPackedBlockArguments (Results 1 – 3 of 3) sorted by relevance
590 unsigned ReplicateOp::GetNumPackedBlockArguments() { in GetNumPackedBlockArguments() function in mlir::tf_device::ReplicateOp596 return GetBody().getNumArguments() - GetNumPackedBlockArguments(); in GetNumReplicatedBlockArguments()602 return GetBody().getArguments().drop_back(GetNumPackedBlockArguments()); in GetReplicatedBlockArguments()608 return GetBody().getArguments().take_back(GetNumPackedBlockArguments()); in GetPackedBlockArguments()
285 unsigned GetNumPackedBlockArguments();
290 new_packed_inputs.reserve(replicate.GetNumPackedBlockArguments()); in AddInputsToReplicateOp()