Searched defs:log_input (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | logging_tensor.py | 139 def log_input(name: str, var: object) -> None: function
|
| H A D | common_modules.py | 460 def poissonnllloss_reference_fn(i, t, log_input=True, full=False, reduction='mean', eps=1e-8): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | loss.h | 792 bool log_input, in poisson_nll_loss()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | loss.cpp | 329 const Tensor& log_input, in forward()
|
| /aosp_15_r20/bionic/benchmarks/ |
| H A D | math_benchmark.cpp | 383 static const std::vector<double> log_input (logf_input.begin(), variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Loss.cpp | 385 Tensor poisson_nll_loss(const Tensor& input, const Tensor& target, const bool log_input, const bool… in poisson_nll_loss()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_impl.py | 46 def log_poisson_loss(targets, log_input, compute_full_loss=False, name=None): argument
|