OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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