Searched refs:TensorPair (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | typed_queue.h | 96 using TensorPair = std::pair<int64_t, Tensor>; variable
99 int64_t SizeOf(const std::priority_queue<TensorPair, U, V>& sq) { in SizeOf() argument
103 return sq.size() * (sizeof(TensorPair) + sq.top().second.AllocatedBytes()); in SizeOf()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe_test.py | 108 TensorPair = collections.namedtuple('TensorPair', ['first', 'second'])
110 pair = TensorPair(a, b)
113 self.assertIsInstance(subscribed, TensorPair)
|
| /external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| D | armnn_runtime.i | 17 %template(TensorPair) pair<int, armnn::Tensor>;
|
| /external/pytorch/test/ |
| D | test_fx.py | 1378 class TensorPair: class
1385 return TensorPair(l, r)
1390 return TensorPair(l, r)
1392 def use_tensor_pair(x : TensorPair, y : TensorPair): argument
1396 x = TensorPair(torch.randn(5, 3), torch.randn(5, 3))
1397 y = TensorPair(torch.randn(5, 3), torch.randn(5, 3))
1408 class TensorPair(metaclass=torch.fx.ProxyableClassMeta): class
1415 return TensorPair(l, r)
1420 return TensorPair(l, r)
1422 def use_tensor_pair_literal(x : TensorPair): argument
[all …]
|