Searched refs:MutableArraySlice (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | range_sampler.h | 53 gtl::MutableArraySlice<int64> batch) const;
76 gtl::MutableArraySlice<int64> batch,
77 gtl::MutableArraySlice<float> batch_expected_count,
79 gtl::MutableArraySlice<float> extras_expected_count) const;
87 gtl::MutableArraySlice<int64> batch,
88 gtl::MutableArraySlice<float> batch_expected_count,
90 gtl::MutableArraySlice<float> extras_expected_count,
128 gtl::MutableArraySlice<int64> batch,
129 gtl::MutableArraySlice<float> batch_expected_count,
131 gtl::MutableArraySlice<float> extras_expected_count,
[all …]
|
| D | range_sampler.cc | 33 using gtl::MutableArraySlice;
38 gtl::MutableArraySlice<int64> batch) const { in SampleBatch()
40 rnd, unique, batch, gtl::MutableArraySlice<float>(), in SampleBatch()
41 gtl::ArraySlice<int64>(), gtl::MutableArraySlice<float>()); in SampleBatch()
45 random::SimplePhilox* rnd, bool unique, gtl::MutableArraySlice<int64> batch, in SampleBatchGetExpectedCount()
46 gtl::MutableArraySlice<float> batch_expected_count, in SampleBatchGetExpectedCount()
48 gtl::MutableArraySlice<float> extras_expected_count) const { in SampleBatchGetExpectedCount()
78 random::SimplePhilox* rnd, bool unique, MutableArraySlice<int64> batch, in SampleBatchGetExpectedCountAvoid()
79 MutableArraySlice<float> batch_expected_count, ArraySlice<int64> extras, in SampleBatchGetExpectedCountAvoid()
80 MutableArraySlice<float> extras_expected_count, in SampleBatchGetExpectedCountAvoid()
[all …]
|
| D | range_sampler_test.cc | 30 using gtl::MutableArraySlice;
286 MutableArraySlice<float>(), all_values, in TEST_F()
295 &rnd, true, absl::MakeSpan(batch), MutableArraySlice<float>(), in TEST_F()
323 &rnd, true, absl::MakeSpan(batch), MutableArraySlice<float>(), in TEST_F()
324 ArraySlice<int64>(), MutableArraySlice<float>(), avoided); in TEST_F()
|
| D | candidate_sampler_ops.cc | 76 gtl::MutableArraySlice<int64> sampled_candidate( in Compute()
78 gtl::MutableArraySlice<float> true_expected_count( in Compute()
81 gtl::MutableArraySlice<float> sampled_expected_count( in Compute()
|
| /external/tensorflow/tensorflow/core/lib/gtl/ |
| D | array_slice.h | 31 using MutableArraySlice = absl::Span<T>; variable
|
| /external/tensorflow/tensorflow/stream_executor/lib/ |
| D | array_slice.h | 27 using MutableArraySlice = absl::Span<T>; variable
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | pmf_to_cdf_op_test.cc | 49 gtl::MutableArraySlice<float> slice) { in GenerateData()
|
| D | pmf_to_cdf_op.cc | 139 gtl::MutableArraySlice<int32> cdf) const { in PerShard()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 295 port::MutableArraySlice<T> host_dst) { in SynchronousMemcpyD2H()
|
| D | stream.h | 686 port::MutableArraySlice<ElementType> host_dst) { in ThenMemcpyD2HQuantized()
1695 port::MutableArraySlice<T> host_dst) { in ThenMemcpyD2H()
|
| D | stream.cc | 169 string ToVlogString(port::MutableArraySlice<T> elements) { in ToVlogString()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 354 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape()
|