def binomial_ml(samples_data, weights_data, out_data): prior_data = qy.stack_allocate(self._model.prior_dtype) qy.value_from_any(1.0).store(prior_data.gep(0, 0)) qy.value_from_any(1.0).store(prior_data.gep(0, 1)) self._map( StridedArray.from_raw(prior_data, (), ()), samples.using(samples_data), weights.using(weights_data), out.using(out_data), )