| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | outfeed_ops.cc | 43 OpKernelContext* ctx, xla::TpuTransferManagerInterface* transfer_manager, in DoWork()
87 OpKernelContext* ctx, xla::TpuTransferManagerInterface* transfer_manager, in DoWork()
|
| D | infeed_ops.cc | 385 xla::TpuTransferManagerInterface* transfer_manager, in DoWork()
418 OpKernelContext* ctx, xla::TpuTransferManagerInterface* transfer_manager, in DoWork()
472 OpKernelContext* ctx, xla::TpuTransferManagerInterface* transfer_manager, in DoWork()
|
| 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
657 xla::TransferManager* const transfer_manager = backend->transfer_manager(); in DoWork() local
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 75 auto transfer_manager = backend->transfer_manager(); in AllocateScopedShapedBuffer() local
173 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
305 auto transfer_manager = backend->transfer_manager(); in SwapIn() local
458 auto transfer_manager = backend->transfer_manager(); in MakeTuple() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 81 xla::TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
|
| D | xla_launch_util.cc | 229 xla::TransferManager* transfer_manager = in PopulateInputs() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.h | 94 TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager() function
|
| D | backend.cc | 123 TransferManager* transfer_manager, in Backend()
|
| D | service.cc | 75 se::Stream* stream, TransferManager* transfer_manager, in RecordArguments()
89 TransferManager* transfer_manager, HloSnapshot* module) { in RecordResult()
|
| /external/crosvm/devices/src/usb/xhci/ |
| D | transfer_ring_controller.rs | 31 transfer_manager: XhciTransferManager, field
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 373 TransferManager* transfer_manager = in AllocateDestinationBuffer() local
663 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostBuffer() local
845 TransferManager* transfer_manager = client()->backend().transfer_manager(); in CreateUninitializedBuffer() local
869 TransferManager* transfer_manager = client()->backend().transfer_manager(); in BufferFromHostLiteral() local
1496 TransferManager* transfer_manager = in MakeTupleHelper() 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 | 955 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
|