Searched defs:parallel_tensor (Results 1 – 4 of 4) sorted by relevance
162 std::unique_ptr<ParallelTensor> parallel_tensor( in ExecuteWithSpecialOps() local233 ParallelTensor* parallel_tensor = reinterpret_cast<ParallelTensor*>(data); in ParallelTensorSummarize() local293 ParallelTensor* parallel_tensor = reinterpret_cast<ParallelTensor*>( in CopyTensorFromParallelDevice() local
130 def _unpack_tensor(self, parallel_tensor): argument140 def unpack(self, parallel_tensor): argument
83 std::unique_ptr<parallel_device::ParallelTensor> parallel_tensor = in BroadcastTensorHandleToParallelTensor() local149 std::unique_ptr<parallel_device::ParallelTensor> parallel_tensor = in BroadcastResourceTensor() local307 std::unique_ptr<parallel_device::ParallelTensor> parallel_tensor = in Broadcast() local
883 std::unique_ptr<parallel_device::ParallelTensor> parallel_tensor = in Pack() local