Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstateless_random_gamma_op_gpu.cu.cc46 FillKernel(int64 num_samples, int64 num_alphas, int64 samples_per_alpha, in FillKernel()
H A Dstateless_random_gamma_op.cc232 const int64_t num_alphas = alpha_t.NumElements(); in Fill() local
H A Drandom_op.cc192 const int64_t num_alphas = alpha_t.NumElements(); in Compute() local