Searched refs:external_reference_hold (Results 1 – 1 of 1) sorted by relevance
121 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference_hold; in AsNumPyArray() member124 TF_ASSIGN_OR_RETURN(hold->external_reference_hold, in AsNumPyArray()127 void* data = hold->external_reference_hold->OpaqueDeviceMemoryDataPointer(); in AsNumPyArray()161 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference_hold, in UnsafeBufferPointer()163 const void* ptr = external_reference_hold->OpaqueDeviceMemoryDataPointer(); in UnsafeBufferPointer()191 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference_hold, in CudaArrayInterface()194 external_reference_hold->OpaqueDeviceMemoryDataPointer(); in CudaArrayInterface()210 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference_hold) in ExtraBufferInfo()211 : external_reference_hold(std::move(external_reference_hold)) {} in ExtraBufferInfo()219 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference_hold; member[all …]