| /external/pytorch/test/distributed/fsdp/ |
| D | test_fsdp_exec_order.py | 69 def get_loss(self, input, output): member in Model
|
| D | test_fsdp_use_orig_params.py | 781 def get_loss(self, inp, out): member in TestFSDPUseOrigParamsParamAccess._test_access_params_after_forward.Model
875 def get_loss(self, inp, out): member in TestFSDPUseOrigParamsWriteback.Model
|
| D | test_fsdp_ignored_modules.py | 66 def get_loss(self, input, output): member in Model
|
| D | test_fsdp_optim_state.py | 168 def get_loss(self, inp, output): member in NestedModel
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_fsdp.py | 95 def get_loss(self, input, output) -> torch.Tensor: member in FSDPTestModel
293 def get_loss(self, input, output): member in TransformerWithSharedParams
424 def get_loss(self, input, output): member in NestedWrappedModule
629 def get_loss(self, input, output): member in ModuleWithDelay
|
| /external/pytorch/benchmarks/functional_autograd_benchmark/ |
| D | torchvision_models.py | 810 def get_loss(self, loss, outputs, targets, indices, num_boxes, **kwargs): member in SetCriterion
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | lookup_ops_test.py | 534 def get_loss(unused_beta): function
559 def get_loss(beta): function
|
| /external/pytorch/test/functorch/ |
| D | test_vmap.py | 242 def get_loss(A): function
|
| /external/pytorch/torch/testing/_internal/distributed/ |
| D | distributed_test.py | 9302 def get_loss(model_output): function
|