ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

oryx.core.interpreters.harvest.HarvestTrace

A HarvestTrace manages HarvestTracer objects.

Since HarvestTracers are just wrappers around known values, HarvestTrace just passes these values through primitives, except in the case of sow and nest, which are specially handled by the active HarvestContext.

Default primitive logic lives in process_primitive, with special logic for sow in handle_sow.

Methods

default_process_primitive

View source

Handles primitives without custom harvest rules.

full_raise

handle_sow

View source

instantiate_const

View source

lift

View source

post_process_call

View source

post_process_map

View source

process_call

View source

process_custom_jvp_call

View source

process_custom_vjp_call

View source

process_higher_order_primitive

View source

process_map

View source

process_primitive

View source

process_xmap

pure

View source

sublift

View source

level Instance of member_descriptor
main Instance of member_descriptor
sublevel Instance of member_descriptor