Home
last modified time | relevance | path

Searched refs:MutableArraySlice (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Drange_sampler.h53 gtl::MutableArraySlice<int64_t> batch) const;
76 gtl::MutableArraySlice<int64_t> batch,
77 gtl::MutableArraySlice<float> batch_expected_count,
79 gtl::MutableArraySlice<float> extras_expected_count) const;
87 gtl::MutableArraySlice<int64_t> batch,
88 gtl::MutableArraySlice<float> batch_expected_count,
90 gtl::MutableArraySlice<float> extras_expected_count,
128 gtl::MutableArraySlice<int64_t> batch,
129 gtl::MutableArraySlice<float> batch_expected_count,
131 gtl::MutableArraySlice<float> extras_expected_count,
[all …]
H A Drange_sampler.cc34 using gtl::MutableArraySlice;
39 gtl::MutableArraySlice<int64_t> batch) const { in SampleBatch()
41 rnd, unique, batch, gtl::MutableArraySlice<float>(), in SampleBatch()
42 gtl::ArraySlice<int64_t>(), gtl::MutableArraySlice<float>()); in SampleBatch()
47 gtl::MutableArraySlice<int64_t> batch, in SampleBatchGetExpectedCount()
48 gtl::MutableArraySlice<float> batch_expected_count, in SampleBatchGetExpectedCount()
50 gtl::MutableArraySlice<float> extras_expected_count) const { in SampleBatchGetExpectedCount()
80 random::SimplePhilox* rnd, bool unique, MutableArraySlice<int64_t> batch, in SampleBatchGetExpectedCountAvoid()
81 MutableArraySlice<float> batch_expected_count, ArraySlice<int64_t> extras, in SampleBatchGetExpectedCountAvoid()
82 MutableArraySlice<float> extras_expected_count, in SampleBatchGetExpectedCountAvoid()
[all …]
H A Drange_sampler_test.cc30 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_t>(), MutableArraySlice<float>(), avoided); in TEST_F()
H A Dcandidate_sampler_ops.cc76 gtl::MutableArraySlice<int64_t> sampled_candidate( in Compute()
78 gtl::MutableArraySlice<float> true_expected_count( in Compute()
81 gtl::MutableArraySlice<float> sampled_expected_count( in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_view.h114 gtl::MutableArraySlice<EdgeInfo> mutable_output_edges() { in mutable_output_edges()
115 return gtl::MutableArraySlice<EdgeInfo>(output_edge_base(), in mutable_output_edges()
/aosp_15_r20/external/tensorflow/tensorflow/core/lib/gtl/
H A Darray_slice.h31 using MutableArraySlice = absl::Span<T>; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/lib/
H A Darray_slice.h27 using MutableArraySlice = absl::Span<T>; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_pimpl.h286 port::MutableArraySlice<T> host_dst) { in SynchronousMemcpyD2H()
H A Dstream.h618 port::MutableArraySlice<ElementType> host_dst) { in ThenMemcpyD2HQuantized()
1183 port::MutableArraySlice<T> host_dst) { in ThenMemcpyD2H()
H A Dstream.cc164 std::string ToVlogString(port::MutableArraySlice<T> elements) { // non-absl ok in ToVlogString()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc524 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape()