Home
last modified time | relevance | path

Searched defs:num_samples_t (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmultinomial_op.cc154 const Tensor& num_samples_t, GuardedPhiloxRandom* generator) { in DoCompute()
230 const Tensor& num_samples_t = ctx->input(1); in Compute() local
286 const Tensor& num_samples_t = ctx->input(1); in Compute() local
H A Dmultinomial_op_test.cc30 Tensor num_samples_t(DT_INT32, TensorShape()); in Multinomial() local