Searched refs:num_alphas (Results 1 – 1 of 1) sorted by relevance
320 const int64 num_alphas = alpha_t.NumElements(); in Compute() local321 OP_REQUIRES(ctx, num_alphas > 0, in Compute()324 num_alphas)); in Compute()327 num_samples * num_alphas, kReservedSamplesPerOutput); in Compute()332 auto DoWork = [num_samples, num_alphas, &rng, samples_flat, alpha_flat]( in Compute()370 samples_alpha_offset[sample_idx * num_alphas] = static_cast<T>(res); in Compute()425 samples_alpha_offset[sample_idx * num_alphas] = in Compute()445 num_alphas * num_samples, kElementCost, DoWork); in Compute()