process_hf_reformer_output() — pytorch Function Reference

Architecture documentation for the process_hf_reformer_output() function in huggingface.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking calls 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ process_hf_reformer_output() — pytorch Function Reference

Dependency Diagram

graph TD
  371b144f_0cc2_e71a_26a0_59e5e0763214["process_hf_reformer_output()"]
  ff1fc40e_5406_a733_9648_4a73bd4fbcb2["AssertionError()"]
  371b144f_0cc2_e71a_26a0_59e5e0763214 -->|calls| ff1fc40e_5406_a733_9648_4a73bd4fbcb2
  style 371b144f_0cc2_e71a_26a0_59e5e0763214 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/huggingface.py lines 83–87

def process_hf_reformer_output(out):
    if not isinstance(out, list):
        raise AssertionError(f"expected out to be a list, got {type(out)}")
    # second output is unstable
    return [elem for i, elem in enumerate(out) if i != 1]