Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dgrpc_tpu_driver.cc267 void AddWriteRequest(std::unique_ptr<StreamRequest::Entry> req) { in AddWriteRequest() function in tpu_driver::__anon4073f9c10111::GrpcTpuStream
730 AddWriteRequest(std::move(req)); in Allocate()
746 AddWriteRequest(std::move(req)); in Allocate()
766 AddWriteRequest(std::move(req)); in AllocateTuple()
779 AddWriteRequest(std::move(req)); in Deallocate()
794 AddWriteRequest(std::move(req)); in TransferToDevice()
812 AddWriteRequest(std::move(req)); in TransferFromDevice()
832 AddWriteRequest(std::move(req)); in TransferFromDeviceToDevice()
857 AddWriteRequest(std::move(req)); in CompileProgram()
879 AddWriteRequest(std::move(req)); in LoadProgram()
[all …]