Searched defs:transfer_manager (Results 1 – 14 of 14) sorted by relevance
76 auto transfer_manager = backend->transfer_manager(); in AllocateScopedShapedBuffer() local174 auto transfer_manager = backend->transfer_manager(); in CreateAndTransfer() local251 auto transfer_manager = backend->transfer_manager(); in StoreToLiteral() local273 auto transfer_manager = backend->transfer_manager(); in WriteLiteral() local306 auto transfer_manager = backend->transfer_manager(); in SwapIn() local459 auto transfer_manager = backend->transfer_manager(); in MakeTuple() local
83 xla::TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
95 TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
122 TransferManager* transfer_manager, in Backend()
79 se::Stream* stream, TransferManager* transfer_manager, in RecordArguments()93 TransferManager* transfer_manager, HloSnapshot* module) { in RecordResult()
383 TransferManager* transfer_manager = in AllocateDestinationBuffer() local617 TransferManager* transfer_manager = in logical_on_device_shape() local713 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostBuffer() local924 TransferManager* transfer_manager = client()->backend().transfer_manager(); in CreateUninitializedBuffer() local948 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostLiteral() local1647 const TransferManager& transfer_manager, in CheckCompatibleShapes()1681 TransferManager* transfer_manager = in MakeTupleHelper() local1774 TransferManager* transfer_manager = in PjRtStreamExecutorExecutable() local1850 TransferManager* transfer_manager = in MakeExecutionInputsAndWaitForEvents() local
33 transfer_manager: XhciTransferManager, field
278 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in BuildComputationInputs() local411 xla::TransferManager* const transfer_manager = in AllocateOutputTensors() local652 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in DoWork() local
161 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in DoTpuExecute() local
150 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in BuildInputBuffers() local
139 auto* transfer_manager = backend->transfer_manager(); in ExecuteAndFetchProfile() local
762 auto* transfer_manager = in BM_DynamicSlice() local
940 auto* transfer_manager = in BM_LocalClientOverhead() local
144 xla::TransferManager* transfer_manager) { in FixTupleTableAsync()