Home
last modified time | relevance | path

Searched defs:TensorPair (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtyped_queue.h96 using TensorPair = std::pair<int64_t, Tensor>; variable
99 int64_t SizeOf(const std::priority_queue<TensorPair, U, V>& sq) { in SizeOf()
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx.py1378 class TensorPair: class
1392 def use_tensor_pair(x : TensorPair, y : TensorPair):
1408 class TensorPair(metaclass=torch.fx.ProxyableClassMeta): class
1422 def use_tensor_pair_literal(x : TensorPair):
1437 class TensorPair(metaclass=torch.fx.ProxyableClassMeta): class
1451 def use_tensor_pair_ctor(x : TensorPair, y : torch.Tensor):