Home
last modified time | relevance | path

Searched defs:TensorFactory (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_legacy_vmap.py885 class TensorFactory: class
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_vmap.py1311 class TensorFactory: class
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py6345 class TensorFactory(torch.nn.Module): class
6357 class TensorFactory(torch.jit.ScriptModule): class
6372 class TensorFactory(torch.jit.ScriptModule): class
6438 class TensorFactory(torch.nn.Module): class