Searched refs:GetReplicaOperandForBlockArgument (Results 1 – 6 of 6) sorted by relevance
84 shape_op.setOperand(replicate_op.GetReplicaOperandForBlockArgument( in MakeShapeOpInvariant()109 replicate_op.GetReplicaOperandForBlockArgument(block_arg, in MakeShapeOpInvariant()
215 replicate.GetReplicaOperandForBlockArgument(block_arg, i), in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()295 replicate.GetReplicaOperandForBlockArgument(arg, i)); in AddInputsToReplicateOp()301 replicate.GetReplicaOperandForBlockArgument(arg, /*replica=*/0)); in AddInputsToReplicateOp()
60 replicate_op.GetReplicaOperandForBlockArgument(arg, i); in runOnOperation()
168 replicate_op.GetReplicaOperandForBlockArgument(block_arg, i)); in ExpandReplicateIntoReplicas()
636 Value ReplicateOp::GetReplicaOperandForBlockArgument(BlockArgument block_arg, in GetReplicaOperandForBlockArgument() function in mlir::tf_device::ReplicateOp
291 Value GetReplicaOperandForBlockArgument(BlockArgument block_arg, unsigned replica);