def _(j):
self._emitters[i].given(
StridedArray.from_typed_pointer(parameter.data.gep(
0, i, j)),
samples.extract(0, i, j),
StridedArray.from_typed_pointer(out.data.gep(0, i, j)),
)
def _(k):
prior_parameter = parameter.at(k).data.gep(0, 1)
posterior_parameter = out.at(k).data.gep(0, 1)
self._sub_emitter.given(
StridedArray.from_typed_pointer(prior_parameter),
samples,
StridedArray.from_typed_pointer(posterior_parameter),
)
def _(k):
prior_parameter = parameter.at(k).data.gep(0, 1)
posterior_parameter = out.at(k).data.gep(0, 1)
self._sub_emitter.given(
StridedArray.from_typed_pointer(prior_parameter),
samples,
StridedArray.from_typed_pointer(posterior_parameter),
)
def _(k):
responsibility = r_KN.at(k, n).data
self._sub_emitter.ll(
StridedArray.from_typed_pointer(out.at(k).data.gep(0, 1)),
StridedArray.from_typed_pointer(sample.data),
responsibility,
)
log_add_double(total.load(), responsibility.load()).store(total)
def _(j):
previous_total = out.load()
self._emitters[i].ll(
StridedArray.from_typed_pointer(parameter.data.gep(0, i, j)),
StridedArray.from_typed_pointer(sample.data.gep(0, i, j) ),
out,
)
(out.load() + previous_total).store(out)
def _(j):
previous_total = out.load()
self._emitters[i].ll(
StridedArray.from_typed_pointer(parameter.data.gep(
0, i, j)),
StridedArray.from_typed_pointer(sample.data.gep(0, i, j)),
out,
)
(out.load() + previous_total).store(out)
def _(k):
self._sub_emitter.ll(
StridedArray.from_typed_pointer(out.at(k).data.gep(0, 1)),
StridedArray.from_typed_pointer(sample.data),
component_ll,
)
log_add_double(
total.load(),
qy.log(out.at(k).data.gep(0, 0).load()) + component_ll.load(),
) \
.store(total)
def _(k):
responsibility = r_KN.at(k, n).data
self._sub_emitter.ll(
StridedArray.from_typed_pointer(
out.at(k).data.gep(0, 1)),
StridedArray.from_typed_pointer(sample.data),
responsibility,
)
log_add_double(total.load(),
responsibility.load()).store(total)
def _(k):
self._sub_emitter.ll(
StridedArray.from_typed_pointer(
out.at(k).data.gep(0, 1)),
StridedArray.from_typed_pointer(sample.data),
component_ll,
)
log_add_double(
total.load(),
qy.log(out.at(k).data.gep(0, 0).load()) + component_ll.load(),
) \
.store(total)
def _(j):
self._emitters[i].map(
priors.extract(0, i, j),
samples.extract(0, i, j),
weights,
StridedArray.from_typed_pointer(out.data.gep(0, i, j)),
)
def _(j):
self._emitters[i].map(
priors.extract(0, i, j),
samples.extract(0, i, j),
weights,
StridedArray.from_typed_pointer(out.data.gep(0, i, j)),
)
def _(k):
j = assigns.at(k).data.load()
self._sub_emitter.ml(
samples.at(j).envelop(),
weights.at(j).envelop(),
StridedArray.from_typed_pointer(out.at(k).data.gep(0, 1)),
)
def _(k):
j = assigns.at(k).data.load()
self._sub_emitter.ml(
samples.at(j).envelop(),
weights.at(j).envelop(),
StridedArray.from_typed_pointer(out.at(k).data.gep(0, 1)),
)
def _(n):
current_pi = posterior_pi.load()
self._sub_emitter.ll(
StridedArray.from_typed_pointer(prior_parameter),
samples.at(n),
posterior_pi,
)
(current_pi + posterior_pi.load()).store(posterior_pi)
def _(k):
component_ll = total.load()
self._sub_emitter.ll(
StridedArray.from_typed_pointer(out.at(k).data.gep(0, 1)),
sample,
total,
)
log_add_double(component_ll, total.load()).store(total)
def _(n):
current_pi = posterior_pi.load()
self._sub_emitter.ll(
StridedArray.from_typed_pointer(prior_parameter),
samples.at(n),
posterior_pi,
)
(current_pi + posterior_pi.load()).store(posterior_pi)
def _(k):
component_ll = total.load()
self._sub_emitter.ll(
StridedArray.from_typed_pointer(
out.at(k).data.gep(0, 1)),
sample,
total,
)
log_add_double(component_ll, total.load()).store(total)
def _(k):
# randomly assign the component
j = qy.random_int(N)
component = StridedArray.from_typed_pointer(out.at(k).data.gep(0, 1))
j.store(assigns.at(k).data)
self._sub_emitter.map(
prior.at(k),
samples.at(j).envelop(),
weights.at(j).envelop(),
component,
)
def _(k):
# randomly assign the component
j = qy.random_int(N)
component = StridedArray.from_typed_pointer(
out.at(k).data.gep(0, 1))
j.store(assigns.at(k).data)
self._sub_emitter.map(
prior.at(k),
samples.at(j).envelop(),
weights.at(j).envelop(),
component,
)
def _(index):
component = parameter.at(index)
self._sub_emitter.ll(
StridedArray.from_typed_pointer(component.data.gep(0, 1)),
sample,
component_ll,
)
log_add_double(
total.load(),
qy.log(component.data.gep(0, 0).load()) + component_ll.load(),
) \
.store(total)
def _(index):
component = parameter.at(index)
self._sub_emitter.ll(
StridedArray.from_typed_pointer(component.data.gep(0, 1)),
sample,
component_ll,
)
log_add_double(
total.load(),
qy.log(component.data.gep(0, 0).load()) + component_ll.load(),
) \
.store(total)
def _(k):
component = out.at(k).data
self._sub_emitter.map(
prior.at(k),
samples,
r_KN.at(k),
StridedArray.from_typed_pointer(component.gep(0, 1)),
)
qy.value_from_any(0.0).store(total)
@qy.for_(N)
def _(n):
(total.load() + r_KN.at(k, n).data.load()).store(total)
(total.load() / float(N)).store(component.gep(0, 0))
def _(k):
component = out.at(k).data
self._sub_emitter.map(
prior.at(k),
samples,
r_KN.at(k),
StridedArray.from_typed_pointer(component.gep(0, 1)),
)
qy.value_from_any(0.0).store(total)
@qy.for_(N)
def _(n):
(total.load() + r_KN.at(k, n).data.load()).store(total)
(total.load() / float(N)).store(component.gep(0, 0))
def _(j):
self._emitters[i].average(
weights,
parameters.extract(0, i, j),
StridedArray.from_typed_pointer(out.data.gep(0, i, j)),
)
def _(j):
self._emitters[i].given(
StridedArray.from_typed_pointer(parameter.data.gep(0, i, j)),
samples.extract(0, i, j),
StridedArray.from_typed_pointer(out.data.gep(0, i, j)),
)
def _(j):
self._emitters[i].average(
weights,
parameters.extract(0, i, j),
StridedArray.from_typed_pointer(out.data.gep(0, i, j)),
)
