Home
last modified time | relevance | path

Searched defs:dst_device_buffer (Results 1 – 3 of 3) sorted by relevance

/external/ml_dtypes/ml_dtypes/tests/
Dintn_test.cc406 DestType* dst_device_buffer = in TYPED_TEST() local
Dfloat8_test.cc1096 DestType* dst_device_buffer = in TYPED_TEST() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc462 std::shared_ptr<TrackedDeviceBuffer> dst_device_buffer = in AllocateDestinationBuffer() local
1431 ScopedHold dst_device_buffer(py_buffer->GetBufferWithUsageHold()); in CopyToDeviceHelper() local