Home
last modified time | relevance | path

Searched refs:WriteSingleTupleIndexTable (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtransfer_manager.cc308 return WriteSingleTupleIndexTable(stream, elements, device_subshape, in WriteTupleIndexTablesAsync()
331 return WriteSingleTupleIndexTable( in WriteRootTupleIndexTable()
351 return WriteSingleTupleIndexTable(stream, elements, buffer_tree.shape(), in WriteRootTupleIndexTable()
H A Dgeneric_transfer_manager.h60 Status WriteSingleTupleIndexTable(
H A Dgeneric_transfer_manager.cc42 Status GenericTransferManager::WriteSingleTupleIndexTable( in WriteSingleTupleIndexTable() function in xla::GenericTransferManager
H A Dtransfer_manager.h307 virtual Status WriteSingleTupleIndexTable(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_transfer_manager.h81 Status WriteSingleTupleIndexTable(
H A Dtpu_transfer_manager.cc270 Status TpuTransferManager::WriteSingleTupleIndexTable( in WriteSingleTupleIndexTable() function in tensorflow::tpu::TpuTransferManager
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_execute.cc162 return transfer_manager->WriteSingleTupleIndexTable( in FixTupleTableAsync()