Searched defs:src_device_buffer (Results 1 – 3 of 3) sorted by relevance
398 std::shared_ptr<TpuSharedBuffer> src_device_buffer = DeviceBuffer(); in CopyToDevice() local
1407 std::shared_ptr<TrackedDeviceBuffer> src_device_buffer) { in CopyToDeviceHelper()1504 ScopedHold src_device_buffer(this, ScopedHold::kUsage); in CopyToDevice() local
1080 TfrtCpuBuffer::ScopedHold src_device_buffer( in CopyToDevice() local