| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/losses/ |
| H A D | losses_impl.py | 179 def compute_loss(losses, weights, loss_collection, reduction): function
572 def compute_loss(labels, predictions, weights, loss_collection): function
|
| /aosp_15_r20/external/pytorch/functorch/benchmarks/ |
| H A D | per_sample_grads.py | 25 def compute_loss(weights, image, target): function
|
| /aosp_15_r20/external/pytorch/functorch/examples/ensembling/ |
| H A D | parallel_train.py | 83 def compute_loss(weights, batch, targets): function
|
| /aosp_15_r20/external/pytorch/functorch/examples/maml_omniglot/ |
| H A D | maml-omniglot-transforms.py | 127 def compute_loss(new_params, buffers, x, y): function
|
| /aosp_15_r20/external/pytorch/functorch/notebooks/_src/ |
| H A D | plot_per_sample_gradients.py | 107 def compute_loss(params, buffers, sample, target): function
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 1646 def compute_loss(weight, x, t): function
1689 def compute_loss(weight, x, t): function
1754 def compute_loss(weights, data, target): function
3844 def compute_loss(params, buffers, x): function
4190 def compute_loss(new_params, buffers, x, y): function
4265 def compute_loss(x, y, params, buffers): function
4400 def compute_loss(weights, batch, targets): function
4504 def compute_loss(weights, batch, targets): function
4560 def compute_loss(weights, image, target): function
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | timm_models.py | 387 def compute_loss(self, pred): member in TimmRunner
|
| H A D | huggingface.py | 513 def compute_loss(self, pred): member in HuggingfaceRunner
|
| H A D | torchbench.py | 433 def compute_loss(self, pred): member in TorchBenchmarkRunner
|