Searched refs:external_reference (Results 1 – 2 of 2) sorted by relevance
47 std::unique_ptr<PjRtBuffer::ExternalReference> external_reference; member272 pack->external_reference = buffer_or.ConsumeValueOrDie(); in BufferToDLPackManagedTensor()273 if (!pack->external_reference) { in BufferToDLPackManagedTensor()282 TF_ASSIGN_OR_RETURN(pack->external_reference, in BufferToDLPackManagedTensor()285 dt.data = pack->external_reference->OpaqueDeviceMemoryDataPointer(); in BufferToDLPackManagedTensor()
367 external_reference = {}368 model.d = external_reference369 external_reference["a"] = []