| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_state.cc | 76 auto transfer_manager = backend->transfer_manager(); in AllocateScopedShapedBuffer() local
174 auto transfer_manager = backend->transfer_manager(); in CreateAndTransfer() local
251 auto transfer_manager = backend->transfer_manager(); in StoreToLiteral() local
273 auto transfer_manager = backend->transfer_manager(); in WriteLiteral() local
306 auto transfer_manager = backend->transfer_manager(); in SwapIn() local
459 auto transfer_manager = backend->transfer_manager(); in MakeTuple() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | pjrt_stream_executor_client.cc | 383 TransferManager* transfer_manager = in AllocateDestinationBuffer() local
617 TransferManager* transfer_manager = in logical_on_device_shape() local
713 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostBuffer() local
924 TransferManager* transfer_manager = client()->backend().transfer_manager(); in CreateUninitializedBuffer() local
948 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostLiteral() local
1647 const TransferManager& transfer_manager, in CheckCompatibleShapes()
1681 TransferManager* transfer_manager = in MakeTupleHelper() local
1774 TransferManager* transfer_manager = in PjRtStreamExecutorExecutable() local
1850 TransferManager* transfer_manager = in MakeExecutionInputsAndWaitForEvents() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_execute_op.cc | 278 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in BuildComputationInputs() local
411 xla::TransferManager* const transfer_manager = in AllocateOutputTensors() local
652 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in DoWork() local
|
| H A D | tpu_reshard_variables_op.cc | 161 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in DoTpuExecute() local
|
| H A D | tpu_reshard_variables_op_util.cc | 150 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in BuildInputBuffers() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_device_context.h | 83 xla::TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | backend.h | 95 TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
|
| H A D | backend.cc | 122 TransferManager* transfer_manager, in Backend()
|
| H A D | service.cc | 79 se::Stream* stream, TransferManager* transfer_manager, in RecordArguments()
93 TransferManager* transfer_manager, HloSnapshot* module) { in RecordResult()
|
| /aosp_15_r20/external/crosvm/devices/src/usb/xhci/ |
| H A D | transfer_ring_controller.rs | 41 transfer_manager: XhciTransferManager, field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | xla_hlo_profile_test.cc | 139 auto* transfer_manager = backend->transfer_manager(); in ExecuteAndFetchProfile() local
|
| H A D | local_client_execute_test.cc | 940 auto* transfer_manager = in BM_LocalClientOverhead() local
|
| H A D | dynamic_ops_test.cc | 762 auto* transfer_manager = in BM_DynamicSlice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_execute.cc | 144 xla::TransferManager* transfer_manager) { in FixTupleTableAsync()
|