| /external/tensorflow/tensorflow/compiler/xrt/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 82 xla::TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
|
| D | xla_launch_util.cc | 242 xla::TransferManager* transfer_manager = in PopulateInputs() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 373 TransferManager* transfer_manager = in AllocateDestinationBuffer() local
607 TransferManager* transfer_manager = in logical_on_device_shape() local
703 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostBuffer() local
914 TransferManager* transfer_manager = client()->backend().transfer_manager(); in CreateUninitializedBuffer() local
938 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostLiteral() local
1601 const TransferManager& transfer_manager, in CheckCompatibleShapes()
1635 TransferManager* transfer_manager = in MakeTupleHelper() local
1728 TransferManager* transfer_manager = in PjRtStreamExecutorExecutable() local
1804 TransferManager* transfer_manager = in MakeExecutionInputsAndWaitForEvents() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 97 TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
|
| D | backend.cc | 123 TransferManager* transfer_manager, in Backend()
|
| D | service.cc | 77 se::Stream* stream, TransferManager* transfer_manager, in RecordArguments()
91 TransferManager* transfer_manager, HloSnapshot* module) { in RecordResult()
|
| /external/crosvm/devices/src/usb/xhci/ |
| D | transfer_ring_controller.rs | 33 transfer_manager: XhciTransferManager, field
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 279 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in BuildComputationInputs() local
412 xla::TransferManager* const transfer_manager = in AllocateOutputTensors() local
653 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in DoWork() local
|
| D | tpu_reshard_variables_op.cc | 161 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in DoTpuExecute() local
|
| D | tpu_reshard_variables_op_util.cc | 150 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in BuildInputBuffers() local
|
| /external/tensorflow/tensorflow/core/platform/s3/ |
| D | s3_file_system.cc | 177 void ShutdownTransferManager(Aws::Transfer::TransferManager* transfer_manager) { in ShutdownTransferManager()
236 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager, in S3RandomAccessFile()
351 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager, in S3WritableFile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 139 auto* transfer_manager = backend->transfer_manager(); in ExecuteAndFetchProfile() local
|
| D | dynamic_ops_test.cc | 759 auto* transfer_manager = in BM_DynamicSlice() local
|
| D | local_client_execute_test.cc | 956 auto* transfer_manager = in BM_LocalClientOverhead() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 142 xla::TransferManager* transfer_manager) { in FixTupleTableAsync()
|
| /external/tensorflow/tensorflow/c/experimental/filesystem/plugins/s3/ |
| D | s3_filesystem.cc | 268 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager; member
373 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager; member
|