Lines Matching defs:num_batches
27 static Graph* PTruncatedNormal(int num_batches, int samples_per_batch) {
30 shape_t.flat<int32>().setValues({num_batches, samples_per_batch});
33 Tensor means_t(DT_FLOAT, TensorShape({num_batches}));
35 Tensor stdevs_t(DT_FLOAT, TensorShape({num_batches}));
38 Tensor minvals_t(DT_FLOAT, TensorShape({num_batches}));
40 Tensor maxvals_t(DT_FLOAT, TensorShape({num_batches}));
56 static Graph* PTruncatedNormal2SD(int num_batches, int samples_per_batch) {
59 shape_t.flat<int32>().setValues({num_batches, samples_per_batch});
61 Tensor means_t(DT_FLOAT, TensorShape({num_batches}));
63 Tensor stdevs_t(DT_FLOAT, TensorShape({num_batches}));
65 Tensor minvals_t(DT_FLOAT, TensorShape({num_batches}));
67 Tensor maxvals_t(DT_FLOAT, TensorShape({num_batches}));
83 static Graph* PTruncatedNormalOneTail(int num_batches, int samples_per_batch) {
86 shape_t.flat<int32>().setValues({num_batches, samples_per_batch});
88 Tensor means_t(DT_FLOAT, TensorShape({num_batches}));
90 Tensor stdevs_t(DT_FLOAT, TensorShape({num_batches}));
92 Tensor minvals_t(DT_FLOAT, TensorShape({num_batches}));
94 Tensor maxvals_t(DT_FLOAT, TensorShape({num_batches}));