Home
last modified time | relevance | path

Searched defs:MultiDeviceBuffer (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dmulti_device_iterator_ops.cc174 class MultiDeviceBuffer { class in tensorflow::data::__anon7578bfd70111::MultiDeviceIterator
176 MultiDeviceBuffer(size_t size, int64_t max_buffer_size, in MultiDeviceBuffer() function in tensorflow::data::__anon7578bfd70111::MultiDeviceIterator::MultiDeviceBuffer