Searched refs:NestedModel (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_meta.py | 95 class NestedModel(nn.Module): class
251 module_regular = NestedModel(device="cuda")
272 module_regular = NestedModel(device="cuda")
293 return NestedModel(device="meta")
305 return NestedModel(device="meta")
320 return deferred_init.deferred_init(NestedModel, device="cuda")
334 return deferred_init.deferred_init(NestedModel, device="cuda")
354 return deferred_init.deferred_init(NestedModel, "cuda")
361 return NestedModel(device="meta")
|
| H A D | test_fsdp_optim_state.py | 144 class NestedModel(torch.nn.Module): class
246 return NestedModel.wrap(model, group), [unmanaged_param]
329 model = NestedModel().to(device)
332 NestedModel.wrap_alt(model, group, fsdp_kwargs)
334 else NestedModel.wrap(model, group, fsdp_kwargs=fsdp_kwargs)
617 model = NestedModel().to(device)
618 wrapped_model = NestedModel.wrap(model, ignore_modules=True)
642 model = NestedModel.wrap(NestedModel().to(device), None)
1195 model = NestedModel().to(device)
1196 model, unmanaged_params = NestedModel.wrap_with_unmanaged_params(
[all …]
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 36 NestedModel,
1162 model = NestedModel().eval()
1183 model = quantize_dynamic(NestedModel().eval(), qconfig_dict)
1186 model = quantize_dynamic(NestedModel().eval(), {'fc3', 'sub2.fc1'}, dtype=dtype)
1194 model = NestedModel().eval()
1216 model = quantize_dynamic(NestedModel().eval(), qconfig_dict, dtype=dtype)
1220 model = quantize_dynamic(NestedModel().eval(), {'fc3', 'sub2'}, dtype=dtype)
1228 model = NestedModel().eval()
1249 model = quantize_dynamic(NestedModel().eval(), qconfig_dynamic_dict)
1253 model = quantize_dynamic(NestedModel().eval(), {'fc3', 'sub2', 'sub2.fc1'}, dtype=dtype)
[all …]
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | modules.cpp | 27 class NestedModel : public torch::nn::Module { class
29 NestedModel() in NestedModel() function in NestedModel
1437 auto model = std::make_shared<NestedModel>(); in TEST_F()
|
| /aosp_15_r20/external/pytorch/test/quantization/jit/ |
| H A D | test_quantize_jit.py | 62 NestedModel,
3734 script_model = NestedModel().eval()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1864 class NestedModel(torch.nn.Module): class
|