Searched defs:transfer_manager (Results 1 – 8 of 8) sorted by relevance
88 auto transfer_manager = backend->transfer_manager(); in AllocateScopedShapedBuffer() local186 auto transfer_manager = backend->transfer_manager(); in CreateAndTransfer() local224 auto transfer_manager = backend->transfer_manager(); in ToLiteral() local249 auto transfer_manager = backend->transfer_manager(); in WriteLiteral() local384 auto transfer_manager = backend->transfer_manager(); in MakeTuple() local
81 xla::TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
94 TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
145 TransferManager* transfer_manager, in Backend()
74 se::Stream* stream, TransferManager* transfer_manager, in RecordArguments()88 TransferManager* transfer_manager, HloSnapshot* module) { in RecordResult()
139 auto* transfer_manager = backend->transfer_manager(); in ExecuteAndFetchProfile() local
742 auto* transfer_manager = in BM_DynamicSlice() local
906 auto* transfer_manager = in BM_LocalClientOverhead() local