digraph { graph [size="12,12"] node [align=left fontsize=12 height=0.2 ranksep=0.1 shape=box style=filled] 5765307696 [label=SqueezeBackward1 fillcolor=darkolivegreen1] 5765307584 -> 5765307696 5765307584 [label=AddmmBackward] 5765308256 -> 5765307584 5765308256 [label="critic.value_heads.value_heads.extrinsic.bias (1)" fillcolor=lightblue] 5765307752 -> 5765307584 5765307752 [label=SumBackward1] 5765482648 -> 5765307752 5765482648 [label=StackBackward] 5765481472 -> 5765482648 5765481472 [label=ReluBackward0] 5765482480 -> 5765481472 5765482480 [label=AddmmBackward] 5765481976 -> 5765482480 5765481976 [label="critic.network_body.vector_encoders.0.seq_layers.1.bias (20)" fillcolor=lightblue] 5765481808 -> 5765482480 5765481808 [label=AddmmBackward] 5765481192 -> 5765481808 5765481192 [label="critic.network_body.vector_encoders.0.seq_layers.0.bias (20)" fillcolor=lightblue] 5765481360 -> 5765481808 5765481360 [label=TBackward] 5765481528 -> 5765481360 5765481528 [label="critic.network_body.vector_encoders.0.seq_layers.0.weight (20, 20)" fillcolor=lightblue] 5765481752 -> 5765482480 5765481752 [label=TBackward] 5765481248 -> 5765481752 5765481248 [label="critic.network_body.vector_encoders.0.seq_layers.1.weight (20, 20)" fillcolor=lightblue] 5765308368 -> 5765307584 5765308368 [label=TBackward] 5765481584 -> 5765308368 5765481584 [label="critic.value_heads.value_heads.extrinsic.weight (1, 20)" fillcolor=lightblue] }