Home
last modified time | relevance | path

Searched defs:samples_flat (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstateless_random_gamma_op_gpu.cu.cc47 random::PhiloxRandom random, T* samples_flat, in FillKernel()
H A Dstateless_random_gamma_op.cc241 auto samples_flat = output->flat<T>().data(); in Fill() local
H A Drandom_poisson_op.cc308 auto samples_flat = samples_t->flat<U>().data(); in Compute() local
H A Dstateless_random_ops.cc193 auto samples_flat = output->flat<U>().data(); in Fill() local
H A Drandom_op.cc197 auto samples_flat = samples_t->flat<T>().data(); in Compute() local