| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/ |
| H A D | gen_tftrt_model.py | 58 class SimpleModel(autotrackable.AutoTrackable): class
97 def SimpleModel(): function
|
| H A D | gen_tf_variablev2_model.py | 41 def SimpleModel(): function
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_tp_integration.py | 53 class SimpleModel(torch.nn.Module): class
89 model: SimpleModel,
|
| H A D | test_fsdp_optim_state.py | 1675 class SimpleModel(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_decorators.py | 97 class SimpleModel(torch.nn.Module): class
151 class SimpleModel(torch.nn.Module): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libtf/tests/ |
| H A D | generate_testdata.py | 58 class SimpleModel(module.Module): class
|
| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/ |
| H A D | test_hsdp_checkpoint.py | 31 class SimpleModel(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_model_dump.py | 19 class SimpleModel(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/ |
| H A D | test_2d_composability.py | 59 class SimpleModel(nn.Module): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/ |
| H A D | quantize_model_test.py | 123 class SimpleModel(module.Module): class in QuantizationMethodTest
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor_compile.py | 53 class SimpleModel(nn.Module): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| H A D | trt_convert_test.py | 104 class SimpleModel(autotrackable.AutoTrackable): class
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | test_vulkan_delegate.py | 650 class SimpleModel(torch.nn.Module): class
666 class SimpleModel(torch.nn.Module): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | graph_constructor_test.cc | 887 TEST_F(GraphConstructorTest, SimpleModel) { in TEST_F() argument
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_dynamo_distributed.py | 776 class SimpleModel(nn.Module): class
|
| /aosp_15_r20/external/executorch/backends/qualcomm/tests/ |
| H A D | models.py | 892 class SimpleModel(torch.nn.Module): class
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestPartitioning.cpp | 1588 TEST_F(PartitioningTest, SimpleModel) { in TEST_F() argument
|