Searched refs:GetNumPackedBlockArguments (Results 1 – 3 of 3) sorted by relevance
602 unsigned ReplicateOp::GetNumPackedBlockArguments() { in GetNumPackedBlockArguments() function in mlir::tf_device::ReplicateOp608 return GetBody().getNumArguments() - GetNumPackedBlockArguments(); in GetNumReplicatedBlockArguments()614 return GetBody().getArguments().drop_back(GetNumPackedBlockArguments()); in GetReplicatedBlockArguments()620 return GetBody().getArguments().take_back(GetNumPackedBlockArguments()); in GetPackedBlockArguments()
287 unsigned GetNumPackedBlockArguments();
243 new_packed_inputs.reserve(replicate.GetNumPackedBlockArguments()); in AddInputsToReplicateOp()