Searched refs:instantiate_const (Results 1 – 3 of 3) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | partial_eval.py | 118 def instantiate_const(self, tracer) -> Tracer: member in JaxprTrace
148 tracers = map(self.instantiate_const, tracers)
530 return trace.instantiate_const(trace.full_raise(tracer))
764 instantiated_tracers = map(trace.instantiate_const, tracers)
1049 def instantiate_const(self, val): member in DynamicJaxprTrace
1076 constvars = map(self.getvar, map(self.instantiate_const, consts))
1106 constvars = map(self.getvar, map(self.instantiate_const, consts))
1131 constvars = map(self.getvar, map(self.instantiate_const, consts))
1152 constvars = map(self.getvar, map(self.instantiate_const, consts))
|
| /dports/math/py-jax/jax-0.2.9/jax/experimental/ |
| H A D | maps.py | 392 constvars = map(self.getvar, map(self.instantiate_const, consts))
|
| /dports/math/py-jax/jax-0.2.9/jax/_src/lax/ |
| H A D | control_flow.py | 897 index_tracer = trace.instantiate_const(tracers[0])
899 ops_tracers = [trace.instantiate_const(t) if uk
1628 new_tracers = [trace.instantiate_const(t) if uk else trace.new_instantiated_literal(core.unit)
|