Searched refs:GetReplicatedBlockArguments (Results 1 – 3 of 3) sorted by relevance
291 for (const auto& arg : replicate.GetReplicatedBlockArguments()) { in AddInputsToReplicateOp()501 BlockArgument state = unformat_replicate.GetReplicatedBlockArguments().back(); in HandleReplicateOp()503 unformat_replicate.GetReplicatedBlockArguments().drop_back(1); in HandleReplicateOp()
601 llvm::ArrayRef<BlockArgument> ReplicateOp::GetReplicatedBlockArguments() { in GetReplicatedBlockArguments() function in mlir::tf_device::ReplicateOp
287 llvm::ArrayRef<BlockArgument> GetReplicatedBlockArguments();