pylibcugraphops.operators.update_efeat_e2e_concat_fwd#

pylibcugraphops.operators.update_efeat_e2e_concat_fwd = <nanobind.nb_func object>#

Computes the forward pass for an operation which concatenates: edge features, and the node features of the source and destination nodes of each each in an edge-to-edge operation.

update_efeat_e2e_concat_fwd(
    output_edge_embedding: device array,
    input_edge_embedding: Optional[device array],
    input_src_node_embedding: Optional[device array],
    input_dst_node_embedding: Optoinal[device array],
    graph: pylibcugraphops.bipartite_csc_int[32|64],
    stream_id: int = 0
)

Parameters:

output_edge_embeddingdevice array type: Device array containing the output edge embeddings.
input_edge_embeddingdevice array type | None: Device array containing the input edge embeddings, ignored if passed as None
input_src_node_embeddingdevice array type | None: Device array containing the input source node embeddings, ignored if passed as None
input_dst_node_embeddingdevice array type | None: Device array containing the input destination node embeddings, ignored if passed as None
graphopaque CSC graph type: graph used for the operation.
stream_idint, default=0: CUDA stream pointer as a python int