Home
last modified time | relevance | path

Searched defs:LinearModel (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/executorch/test/models/
H A Dlinear_model.py10 class LinearModel(torch.nn.Module): class
/aosp_15_r20/external/executorch/build/packaging/
H A Dsmoke_test.py41 class LinearModel(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime_cuda.py99 class LinearModel(torch.nn.Module): class
H A Dtest_pytorch_onnx_onnxruntime.py5632 class LinearModel(torch.nn.Module): class
5645 class LinearModel(torch.nn.Module): class
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_strategy_model_parallelism_test.py308 class LinearModel(module.Module): class
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py3018 class LinearModel(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py5750 class LinearModel(torch.nn.Module): class
6683 class LinearModel(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_aot_inductor.py426 class LinearModel(torch.nn.Module): class