Searched refs:sample_idx (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_poisson_op.cc | 134 for (int64 sample_idx = output_idx % num_samples; in operator ()() local
135 sample_idx < num_samples && output_idx < limit_output; in operator ()()
136 sample_idx++, output_idx++) { in operator ()()
151 samples_rate_output[sample_idx * num_rate] = U(x); in operator ()()
196 for (int64 sample_idx = output_idx % num_samples; in operator ()() local
197 sample_idx < num_samples && output_idx < limit_output; in operator ()()
198 sample_idx++, output_idx++) { in operator ()()
223 samples_rate_output[sample_idx * num_rate] = U(k); in operator ()()
236 samples_rate_output[sample_idx * num_rate] = U(k); in operator ()()
|
| D | random_op.cc | 360 for (int64 sample_idx = output_idx % num_samples; in Compute() local
361 sample_idx < num_samples && output_idx < limit_output; in Compute()
362 sample_idx++, output_idx++) { in Compute()
370 samples_alpha_offset[sample_idx * num_alphas] = static_cast<T>(res); in Compute()
387 for (int64 sample_idx = output_idx % num_samples; in Compute() local
388 sample_idx < num_samples && output_idx < limit_output; in Compute()
389 sample_idx++, output_idx++) { in Compute()
425 samples_alpha_offset[sample_idx * num_alphas] = in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1510 llvm::Value* sample_idx, llvm::Value* hlo_random_value, in CalculateSampleValues() argument
1513 llvm::Type* index_ty = sample_idx->getType(); in CalculateSampleValues()
1521 counter_values[0] = sample_idx; in CalculateSampleValues()
1525 llvm_ir::SplitInt64ToInt32s(b, sample_idx); in CalculateSampleValues()
1641 llvm::Value* sample_idx = UDiv(elem_idx, elems_per_sample_value); in MakePhiloxRngElementGenerator() local
1645 sample_idx, hlo_random_value, global_random_number, rng_state, b_); in MakePhiloxRngElementGenerator()
|