Searched refs:IsReplicatedBlockArgument (Results 1 – 3 of 3) sorted by relevance
201 if (replicate.IsReplicatedBlockArgument(block_arg)) { in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()314 if (replicate.IsReplicatedBlockArgument(arg)) { in AddInputsToReplicateOp()
612 bool ReplicateOp::IsReplicatedBlockArgument(BlockArgument block_arg) { in IsReplicatedBlockArgument() function in mlir::tf_device::ReplicateOp619 return !IsReplicatedBlockArgument(block_arg); in IsPackedBlockArgument()
288 bool IsReplicatedBlockArgument(BlockArgument block_arg);