Searched refs:samples_per_batch (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | parameterized_truncated_normal_op.cc | 53 int64 samples_per_batch, int64 num_elements, in operator ()()
70 auto DoWork = [samples_per_batch, num_elements, &ctx, &means, &stddevs, in operator ()()
81 gen_copy.Skip(start_batch * 2 * kMaxIterations * (samples_per_batch + 3) / in operator ()()
105 std::min((b + 1) * samples_per_batch, num_elements); in operator ()()
106 int64 sample = b * samples_per_batch; in operator ()()
306 batchInitCost + uniformRejectionSamplingCost * 2 * samples_per_batch; in operator ()()
341 int32 samples_per_batch = 1; in Compute() local
344 samples_per_batch *= shape_tensor.flat<int32>()(i); in Compute()
346 const int32 num_elements = num_batches * samples_per_batch; in Compute()
380 int32 size = num_batches * samples_per_batch; in Compute()
[all …]
|
| D | parameterized_truncated_normal_op_test.cc | 27 static Graph* PTruncatedNormal(int num_batches, int samples_per_batch) { in PTruncatedNormal() argument
30 shape_t.flat<int32>().setValues({num_batches, samples_per_batch}); in PTruncatedNormal()
56 static Graph* PTruncatedNormal2SD(int num_batches, int samples_per_batch) { in PTruncatedNormal2SD() argument
59 shape_t.flat<int32>().setValues({num_batches, samples_per_batch}); in PTruncatedNormal2SD()
83 static Graph* PTruncatedNormalOneTail(int num_batches, int samples_per_batch) { in PTruncatedNormalOneTail() argument
86 shape_t.flat<int32>().setValues({num_batches, samples_per_batch}); in PTruncatedNormalOneTail()
|
| D | parameterized_truncated_normal_op_gpu.cu.cc | 55 int64 samples_per_batch, int64 num_elements, in TruncatedNormalKernel() argument
92 const int64 batch_id = offset / samples_per_batch; in TruncatedNormalKernel()
236 int64 samples_per_batch, int64 num_elements, in operator ()()
247 0, d.stream(), gen, output.data(), num_batches, samples_per_batch, in operator ()()
|
| D | parameterized_truncated_normal_op.h | 38 int64 samples_per_batch, int64 num_elements,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ParameterizedTruncatedNormal.pbtxt | 37 A matrix of shape num_batches x samples_per_batch, filled with random
|