Home
last modified time | relevance | path

Searched defs:sample_idx (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drandom_poisson_op.cc134 for (int64 sample_idx = output_idx % num_samples; in operator ()() local
196 for (int64 sample_idx = output_idx % num_samples; in operator ()() local
Drandom_op.cc360 for (int64 sample_idx = output_idx % num_samples; in Compute() local
387 for (int64 sample_idx = output_idx % num_samples; in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1510 llvm::Value* sample_idx, llvm::Value* hlo_random_value, in CalculateSampleValues()
1641 llvm::Value* sample_idx = UDiv(elem_idx, elems_per_sample_value); in MakePhiloxRngElementGenerator() local