Searched defs:samples_per_batch (Results 1 – 3 of 3) sorted by relevance
27 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()
55 int64 samples_per_batch, int64 num_elements, in TruncatedNormalKernel()
341 int32 samples_per_batch = 1; in Compute() local