Searched refs:IsReplicatedBlockArgument (Results 1 – 3 of 3) sorted by relevance
154 if (replicate.IsReplicatedBlockArgument(block_arg)) { in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()267 if (replicate.IsReplicatedBlockArgument(arg)) { in AddInputsToReplicateOp()
624 bool ReplicateOp::IsReplicatedBlockArgument(BlockArgument block_arg) { in IsReplicatedBlockArgument() function in mlir::tf_device::ReplicateOp631 return !IsReplicatedBlockArgument(block_arg); in IsPackedBlockArgument()
290 bool IsReplicatedBlockArgument(BlockArgument block_arg);