HomeSort by relevance Sort by last modified time
    Searched full:samples_per_batch (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
parameterized_truncated_normal_op.cc 53 int64 samples_per_batch, int64 num_elements,
62 auto DoWork = [samples_per_batch, num_elements, &ctx, &means, &stddevs,
72 gen_copy.Skip(start_batch * 2 * kMaxIterations * (samples_per_batch + 3) /
92 // samples_per_batch.
94 std::min((b + 1) * samples_per_batch, num_elements);
95 int64 sample = b * samples_per_batch;
231 batchInitCost + uniformRejectionSamplingCost * 2 * samples_per_batch;
266 int32 samples_per_batch = 1; variable
269 samples_per_batch *= shape_tensor.flat<int32>()(i);
271 const int32 num_elements = num_batches * samples_per_batch;
    [all...]
parameterized_truncated_normal_op_test.cc 27 static Graph* PTruncatedNormal(int num_batches, int samples_per_batch) {
30 shape_t.flat<int32>().setValues({num_batches, samples_per_batch});
56 static Graph* PTruncatedNormal2SD(int num_batches, int samples_per_batch) {
59 shape_t.flat<int32>().setValues({num_batches, samples_per_batch});
83 static Graph* PTruncatedNormalOneTail(int num_batches, int samples_per_batch) {
86 shape_t.flat<int32>().setValues({num_batches, samples_per_batch});
parameterized_truncated_normal_op_gpu.cu.cc 52 int64 samples_per_batch, int64 num_elements,
79 const int64 batch_id = offset / samples_per_batch;
196 int64 samples_per_batch, int64 num_elements,
207 gen, output.data(), num_batches, samples_per_batch, num_elements,
parameterized_truncated_normal_op.h 38 int64 samples_per_batch, int64 num_elements,
  /external/tensorflow/tensorflow/core/api_def/base_api/
api_def_ParameterizedTruncatedNormal.pbtxt 37 A matrix of shape num_batches x samples_per_batch, filled with random

Completed in 95 milliseconds