Searched refs:TensorPair (Results 1 – 2 of 2) sorted by relevance
98 using TensorPair = std::pair<int64, PersistentTensor>; variable101 int64 SizeOf(const std::priority_queue<TensorPair, U, V>& sq) { in SizeOf() argument105 return sq.size() * (sizeof(TensorPair) + sq.top().second.AllocatedBytes()); in SizeOf()
112 TensorPair = collections.namedtuple('TensorPair', ['first', 'second'])114 pair = TensorPair(a, b)117 self.assertIsInstance(subscribed, TensorPair)