Home
last modified time | relevance | path

Searched refs:sample_idx (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drandom_binomial_op.cc219 for (int64 sample_idx = output_idx % samples_per_batch; in operator ()() local
220 sample_idx < samples_per_batch && output_idx < limit_output; in operator ()()
221 ++sample_idx, ++output_idx) { in operator ()()
222 output_batch_offset[sample_idx * num_batches] = static_cast<U>(0.0); in operator ()()
225 for (int64 sample_idx = output_idx % samples_per_batch; in operator ()() local
226 sample_idx < samples_per_batch && output_idx < limit_output; in operator ()()
227 ++sample_idx, ++output_idx) { in operator ()()
228 output_batch_offset[sample_idx * num_batches] = in operator ()()
234 for (int64 sample_idx = output_idx % samples_per_batch; in operator ()() local
235 sample_idx < samples_per_batch && output_idx < limit_output; in operator ()()
[all …]
Drandom_poisson_op.cc128 for (int64 sample_idx = output_idx % num_samples; in operator ()() local
129 sample_idx < num_samples && output_idx < limit_output; in operator ()()
130 sample_idx++, output_idx++) { in operator ()()
145 samples_rate_output[sample_idx * num_rate] = U(x); in operator ()()
155 for (int64 sample_idx = output_idx % num_samples; in operator ()() local
156 sample_idx < num_samples && output_idx < limit_output; in operator ()()
157 sample_idx++, output_idx++) { in operator ()()
159 samples_rate_output[sample_idx * num_rate] = k; in operator ()()
200 for (int64 sample_idx = output_idx % num_samples; in operator ()() local
201 sample_idx < num_samples && output_idx < limit_output; in operator ()()
[all …]
Dstateless_random_gamma_op.cc99 for (int64 sample_idx = output_idx % samples_per_alpha; in Fill() local
100 sample_idx < samples_per_alpha && output_idx < limit_output; in Fill()
101 sample_idx++, output_idx++) { in Fill()
108 samples_alpha_offset[sample_idx * num_alphas] = static_cast<T>(res); in Fill()
125 for (int64 sample_idx = output_idx % samples_per_alpha; in Fill() local
126 sample_idx < samples_per_alpha && output_idx < limit_output; in Fill()
127 sample_idx++, output_idx++) { in Fill()
164 samples_alpha_offset[sample_idx * num_alphas] = in Fill()
Dparameterized_truncated_normal_op.cc434 for (int64 sample_idx = output_idx % samples_per_batch; in operator ()() local
435 sample_idx < samples_per_batch && output_idx < limit_output;) { in operator ()()
441 output_batch_offset[sample_idx * num_batches] = in operator ()()
443 ++sample_idx; in operator ()()
445 if (sample_idx >= samples_per_batch || in operator ()()
475 for (int64 sample_idx = output_idx % samples_per_batch; in operator ()() local
476 sample_idx < samples_per_batch && output_idx < limit_output;) { in operator ()()
506 output_batch_offset[sample_idx * num_batches] = in operator ()()
508 ++sample_idx; in operator ()()
510 if (sample_idx >= samples_per_batch || in operator ()()
[all …]
Drandom_op.cc233 for (int64 sample_idx = output_idx % samples_per_alpha; in Compute() local
234 sample_idx < samples_per_alpha && output_idx < limit_output; in Compute()
235 sample_idx++, output_idx++) { in Compute()
243 samples_alpha_offset[sample_idx * num_alphas] = static_cast<T>(res); in Compute()
260 for (int64 sample_idx = output_idx % samples_per_alpha; in Compute() local
261 sample_idx < samples_per_alpha && output_idx < limit_output; in Compute()
262 sample_idx++, output_idx++) { in Compute()
298 samples_alpha_offset[sample_idx * num_alphas] = in Compute()
Dstateless_random_gamma_op_gpu.cu.cc65 int64 sample_idx = output_idx % samples_per_alpha; in FillKernel() local
82 samples_flat[alpha_idx + sample_idx * num_alphas] = static_cast<T>(res); in FillKernel()
136 samples_flat[alpha_idx + sample_idx * num_alphas] = in FillKernel()
/external/autotest/client/cros/power/
Dpower_telemetry_utils.py91 sample_idx = numpy.arange(len(data))[[~nan_data]]
93 if not len(sample_idx):
96 output = numpy.interp(range(len(data)), sample_idx, sample_vals)
/external/skia/modules/skottie/src/effects/
DMotionBlurEffect.cpp70 SkRect MotionBlurEffect::seekToSample(size_t sample_idx, const SkMatrix& ctm) const { in seekToSample() argument
71 SkASSERT(sample_idx < fSampleCount); in seekToSample()
72 fAnimator->seek(fT + fPhase + fDT * sample_idx); in seekToSample()
DMotionBlurEffect.h39 SkRect seekToSample(size_t sample_idx, const SkMatrix& ctm) const;
/external/mesa3d/src/broadcom/compiler/
Dnir_to_vir.c2122 unsigned sample_idx = nir_intrinsic_base(instr); in ntq_emit_per_sample_color_write() local
2123 assert(sample_idx < V3D_MAX_SAMPLES); in ntq_emit_per_sample_color_write()
2125 unsigned offset = (rt * V3D_MAX_SAMPLES + sample_idx) * 4; in ntq_emit_per_sample_color_write()
2221 ntq_get_sample_offset(struct v3d_compile *c, struct qreg sample_idx, in ntq_get_sample_offset() argument
2224 sample_idx = vir_ITOF(c, sample_idx); in ntq_get_sample_offset()
2228 vir_FMUL(c, sample_idx, in ntq_get_sample_offset()
2231 vir_uniform_f(c, 2.0f), sample_idx), in ntq_get_sample_offset()
2239 vir_FMUL(c, sample_idx, in ntq_get_sample_offset()
2283 struct qreg sample_idx = i3; in ntq_get_barycentric_centroid() local
2285 sample_idx = vir_SEL(c, V3D_QPU_COND_IFNA, i1, sample_idx); in ntq_get_barycentric_centroid()
[all …]
/external/mesa3d/src/amd/vulkan/
Dradv_meta_blit2d.c502 nir_intrinsic_instr *sample_idx = NULL; in build_nir_texel_fetch() local
519 sample_idx = nir_intrinsic_instr_create(b->shader, nir_intrinsic_load_sample_id); in build_nir_texel_fetch()
520 nir_ssa_dest_init(&sample_idx->instr, &sample_idx->dest, 1, 32, "sample_idx"); in build_nir_texel_fetch()
521 nir_builder_instr_insert(b, &sample_idx->instr); in build_nir_texel_fetch()
532 tex->src[1].src = nir_src_for_ssa(is_multisampled ? &sample_idx->dest.ssa : nir_imm_int(b, 0)); in build_nir_texel_fetch()
/external/mesa3d/src/broadcom/vulkan/
Dv3dv_meta_copy.c3382 nir_ssa_def *sample_idx) in build_nir_tex_op_ms_fetch_sample() argument
3394 tex->src[3].src = nir_src_for_ssa(sample_idx); in build_nir_tex_op_ms_fetch_sample()