View source on GitHub
|
A HarvestTrace manages HarvestTracer objects.
oryx.core.interpreters.harvest.HarvestTrace(
main: "MainTrace",
sublevel: "Sublevel"
) -> None
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
full_raise
full_raise(
val
) -> "Tracer"
handle_sow
handle_sow(
name, *tracers, tag, mode, tree
)
instantiate_const
instantiate_const(
val
)
lift
lift(
val
)
post_process_call
post_process_call(
call_primitive, out_tracers, params
)
post_process_map
post_process_map(
call_primitive, out_tracers, params
)
process_call
process_call(
call_primitive, f, tracers, params
)
process_custom_jvp_call
process_custom_jvp_call(
primitive, fun, jvp, tracers
)
process_custom_vjp_call
process_custom_vjp_call(
primitive, fun, fwd, bwd, tracers, out_trees
)
process_higher_order_primitive
process_higher_order_primitive(
primitive, f, tracers, params, is_map
)
process_map
process_map(
call_primitive, f, tracers, params
)
process_primitive
process_primitive(
primitive, tracers, params
)
pure
pure(
val
)
sublift
sublift(
tracer
)