Searched refs:GetPackedBlockArguments (Results 1 – 4 of 4) sorted by relevance
66 for (auto composite_arguments : replicate.GetPackedBlockArguments()) { in GetResourceOpsUsingCompositeArgsInReplicate()
299 for (const auto& arg : replicate.GetPackedBlockArguments()) { in AddInputsToReplicateOp()504 auto packed_block_args = unformat_replicate.GetPackedBlockArguments(); in HandleReplicateOp()
607 llvm::ArrayRef<BlockArgument> ReplicateOp::GetPackedBlockArguments() { in GetPackedBlockArguments() function in mlir::tf_device::ReplicateOp
286 llvm::ArrayRef<BlockArgument> GetPackedBlockArguments();