Searched refs:new_instantiated_const (Results 1 – 2 of 2) sorted by relevance
108 def new_instantiated_const(self, val) -> 'JaxprTracer': member in JaxprTrace126 return self.new_instantiated_const(const)216 const_tracers = map(self.new_instantiated_const, consts)263 const_tracers = map(trace.new_instantiated_const, consts)317 const_tracers = map(self.new_instantiated_const, consts)356 const_tracers = map(self.new_instantiated_const, consts)780 const_tracers = map(trace.new_instantiated_const, consts)
903 res_tracers = _map(trace.new_instantiated_const, res)1636 int_res_tracers = _map(trace.new_instantiated_const, intensive_residuals)1637 ext_res_tracers = _map(trace.new_instantiated_const, extensive_residuals)