Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/overrides_benchmark/
H A Dpyspybench.py24 TensorClass = globals()[args.tensor_class] variable
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py144 TensorClass = namedtuple("TensorClass", ["tensors"]) variable