Searched refs:tpAllocation (Results 1 – 2 of 2) sorted by relevance
214 tensorpipe::Allocation tpAllocation; in tensorpipeAllocate() local222 tpAllocation.payloads.resize(tpDescriptor.payloads.size()); in tensorpipeAllocate()232 tpAllocation.payloads[kTpMessageTypeIdx].data = buffers.type.get(); in tensorpipeAllocate()242 tpAllocation.payloads[kTpMessageIdIdx].data = buffers.id.get(); in tensorpipeAllocate()246 tpAllocation.payloads[kTpMessagePayloadIdx].data = buffers.payload.data(); in tensorpipeAllocate()249 tpAllocation.payloads[kTpMessagePickleIdx].data = buffers.pickle.data(); in tensorpipeAllocate()252 tpAllocation.tensors.reserve(numTensors); in tensorpipeAllocate()267 TORCH_INTERNAL_ASSERT(tpAllocation.tensors.size() == tensorIdx); in tensorpipeAllocate()275 tpAllocation); in tensorpipeAllocate()276 TORCH_INTERNAL_ASSERT(tpAllocation.tensors.size() == tensorIdx + 1); in tensorpipeAllocate()[all …]
563 auto [tpAllocation, tpBuffers] = tensorpipeAllocate(tpDescriptor, streams); in pipeRead()566 std::move(tpAllocation), in pipeRead()