Home
last modified time | relevance | path

Searched defs:src_device_buffer (Results 1 – 5 of 5) sorted by relevance

/external/ml_dtypes/ml_dtypes/tests/
Dintn_test.cc405 IntN* src_device_buffer = (IntN*)device.allocate(kNumElems * sizeof(IntN)); in TYPED_TEST() local
Dfloat8_test.cc1094 Float8* src_device_buffer = in TYPED_TEST() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc408 std::shared_ptr<TpuSharedBuffer> src_device_buffer = DeviceBuffer(); in CopyToDevice() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1420 std::shared_ptr<TrackedDeviceBuffer> src_device_buffer) { in CopyToDeviceHelper()
1517 ScopedHold src_device_buffer(this, ScopedHold::kUsage); in CopyToDevice() local
Dtfrt_cpu_pjrt_client.cc1044 auto* src_device_buffer = AcquireUsage(usage_event); in CopyToDevice() local