Searched refs:batch_expected_count (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | range_sampler.cc | 47 gtl::MutableArraySlice<float> batch_expected_count, in SampleBatchGetExpectedCount() argument
50 SampleBatchGetExpectedCountAvoid(rnd, unique, batch, batch_expected_count, in SampleBatchGetExpectedCount()
80 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid() argument
109 if (!batch_expected_count.empty()) { in SampleBatchGetExpectedCountAvoid()
110 CHECK_EQ(batch_size, batch_expected_count.size()); in SampleBatchGetExpectedCountAvoid()
112 batch_expected_count[i] = in SampleBatchGetExpectedCountAvoid()
127 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid() argument
135 if (!batch_expected_count.empty()) { in SampleBatchGetExpectedCountAvoid()
136 CHECK_EQ(batch_size, batch_expected_count.size()); in SampleBatchGetExpectedCountAvoid()
138 batch_expected_count[i] = 1; in SampleBatchGetExpectedCountAvoid()
[all …]
|
| D | range_sampler.h | 77 gtl::MutableArraySlice<float> batch_expected_count,
88 gtl::MutableArraySlice<float> batch_expected_count,
129 gtl::MutableArraySlice<float> batch_expected_count,
194 gtl::MutableArraySlice<float> batch_expected_count,
|