Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_executable_interface.cc40 TF_RETURN_IF_ERROR(transfer_manager->WriteTupleIndexTablesAsync( in PopulateResultTupleBuffers()
47 return transfer_manager->WriteTupleIndexTablesAsync(stream, result); in PopulateResultTupleBuffers()
/external/tensorflow/tensorflow/compiler/xla/service/
Dgeneric_transfer_manager.cc110 TF_RETURN_IF_ERROR(WriteTupleIndexTablesAsync(stream, device_buffer)); in TransferLiteralToDeviceAsync()
Dtransfer_manager.cc283 TF_RETURN_IF_ERROR(WriteTupleIndexTablesAsync(stream, device_buffer)); in WriteTupleIndexTables()
287 Status TransferManager::WriteTupleIndexTablesAsync( in WriteTupleIndexTablesAsync() function in xla::TransferManager
Dtransfer_manager.h218 Status WriteTupleIndexTablesAsync(se::Stream* stream,
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tpu_device.cc256 dst_xla_context->transfer_manager()->WriteTupleIndexTablesAsync( in TpuDeviceToDeviceCopy()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc261 transfer_manager->WriteTupleIndexTablesAsync(stream, shaped_buffer)); in UpdateDynamicInputs()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc445 TF_RETURN_IF_ERROR(transfer_manager->WriteTupleIndexTablesAsync( in AllocateDestinationBuffer()