Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Drandom_binomial_op_test.cc28 int samples_per_batch) { in RandomBinomialGraph()
48 static Graph* RandomBinomialInv(int num_batches, int samples_per_batch) { in RandomBinomialInv()
53 static Graph* RandomBinomialRej(int num_batches, int samples_per_batch) { in RandomBinomialRej()
59 int samples_per_batch) { in RandomBinomialInvComplement()
65 int samples_per_batch) { in RandomBinomialRejComplement()
Dparameterized_truncated_normal_op_test.cc27 static Graph* PTruncatedNormal(int num_batches, int samples_per_batch) { in PTruncatedNormal()
56 static Graph* PTruncatedNormal2SD(int num_batches, int samples_per_batch) { in PTruncatedNormal2SD()
83 static Graph* PTruncatedNormalOneTail(int num_batches, int samples_per_batch) { in PTruncatedNormalOneTail()
Drandom_binomial_op.cc381 int64 samples_per_batch = 1; in Compute() local
493 int64 samples_per_batch = 1; in Compute() local
Dparameterized_truncated_normal_op.cc632 int32 samples_per_batch = 1; in Compute() local
786 int64 samples_per_batch = 1; in Compute() local
Dparameterized_truncated_normal_op_gpu.cu.cc55 int64 samples_per_batch, int64 num_elements, in TruncatedNormalKernel()