Searched refs:batch_expected_count (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | range_sampler.cc | 46 gtl::MutableArraySlice<float> batch_expected_count, in SampleBatchGetExpectedCount() argument
49 SampleBatchGetExpectedCountAvoid(rnd, unique, batch, batch_expected_count, in SampleBatchGetExpectedCount()
79 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid() argument
108 if (!batch_expected_count.empty()) { in SampleBatchGetExpectedCountAvoid()
109 CHECK_EQ(batch_size, batch_expected_count.size()); in SampleBatchGetExpectedCountAvoid()
111 batch_expected_count[i] = in SampleBatchGetExpectedCountAvoid()
126 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid() argument
134 if (!batch_expected_count.empty()) { in SampleBatchGetExpectedCountAvoid()
135 CHECK_EQ(batch_size, batch_expected_count.size()); in SampleBatchGetExpectedCountAvoid()
137 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,
|