Home
last modified time | relevance | path

Searched refs:LinearModel (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/executorch/build/packaging/
H A Dsmoke_test.py41 class LinearModel(torch.nn.Module): class
68 exported_program = export(LinearModel(), example_inputs)
/aosp_15_r20/external/executorch/test/models/
H A Dgenerate_linear_out_bundled_program.py30 from executorch.test.models.linear_model import LinearModel
35 model = LinearModel()
H A Dlinear_model.py10 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
112 model = amp.initialize(LinearModel(), opt_level="O2")
H A Dtest_pytorch_onnx_onnxruntime.py5632 class LinearModel(torch.nn.Module): class
5643 self.run_test(LinearModel(), (x,))
5645 class LinearModel(torch.nn.Module): class
5653 self.run_test(LinearModel(), (x, y, z))
5659 self.run_test(LinearModel(), (x, y, z))
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_strategy_model_parallelism_test.py308 class LinearModel(module.Module): class
311 super(LinearModel, self).__init__()
332 model = LinearModel(w1)
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_aot_inductor.py426 class LinearModel(torch.nn.Module): class
438 self.check_model(LinearModel(self.device), example_inputs)
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py3018 class LinearModel(torch.nn.Module): class
3029 model = LinearModel().eval()
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py5750 class LinearModel(torch.nn.Module): class
5759 m = LinearModel()
6683 class LinearModel(torch.nn.Module): class
6719 model = LinearModel()
/aosp_15_r20/external/tensorflow/
H A DRELEASE.md1191 * `LinearModel` and `WideDeepModel` are moved to the
1193 (`tf.compat.v1.keras.models.LinearModel` and
1195 endpoints (`tf.keras.experimental.models.LinearModel` and