Searched refs:parallel_tensor (Results 1 – 2 of 2) sorted by relevance
154 std::unique_ptr<ParallelTensor> parallel_tensor( in ExecuteWithSpecialOps() local158 parallel_inputs.push_back(parallel_tensor.get()); in ExecuteWithSpecialOps()159 implicitly_broadcast_tensors.emplace_back(std::move(parallel_tensor)); in ExecuteWithSpecialOps()242 std::unique_ptr<ParallelTensor> parallel_tensor( in CopyToParallelDevice() local246 std::move(parallel_tensor), status) in CopyToParallelDevice()
96 def unpack(self, parallel_tensor): argument108 parallel_tensor, num_replicas=len(self.components))