Home
last modified time | relevance | path

Searched defs:loss (Results 1 – 25 of 116) sorted by relevance

12345

/external/tensorflow/tensorflow/python/training/
Doptimizer_test.py46 def loss(): function
121 def loss(): function
142 def loss(): function
159 def loss(): function
189 def loss(): function
Dmomentum_test.py245 def loss(): function
274 def loss(): function
/external/tensorflow/tensorflow/python/keras/saving/utils_v1/
Dsignature_def_utils.py24 inputs, loss, predictions=None, metrics=None): argument
31 inputs, loss, predictions=None, metrics=None): argument
38 method_name, inputs, loss=None, predictions=None, argument
Dexport_output.py246 def __init__(self, loss=None, predictions=None, metrics=None): argument
360 def loss(self): member in _SupervisedOutput
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py58 def get_gradients(opt, loss, params): argument
122 loss = lambda: grad_check_fn(var) / strategy.num_replicas_in_sync function
209 loss = lambda: var * 2.0 / strategy.num_replicas_in_sync function
220 loss = lambda: var * float('NaN') function
243 loss = lambda: var / strategy.num_replicas_in_sync function
Dloss_scale_optimizer.py81 loss, argument
128 def _scale_loss(self, loss): argument
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py77 def get_updates(self, loss, params): argument
80 def get_gradients(self, loss, params): argument
191 def get_updates(self, loss, params): argument
265 def get_updates(self, loss, params): argument
340 def get_updates(self, loss, params): argument
423 def get_updates(self, loss, params): argument
521 def get_updates(self, loss, params): argument
617 def get_updates(self, loss, params): argument
708 def get_updates(self, loss, params): argument
782 def minimize(self, loss, var_list, grad_loss=None, tape=None): argument
[all …]
/external/libopus/dnn/training_tf2/
Dlossfuncs.py12 def loss(y_true,y_pred): function
26 def loss(y_true,y_pred): function
93 def loss(y_true,y_pred): function
/external/tensorflow/tensorflow/python/saved_model/
Dsignature_def_utils_impl.py217 inputs, loss, predictions=None, metrics=None): argument
224 inputs, loss, predictions=None, metrics=None): argument
231 method_name, inputs, loss=None, predictions=None, argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc90 auto loss = XlaHelpers::ConvertElementType(sum, type); in CrossEntropyWithLogits() local
113 xla::XlaOp loss, backprop; in Compile() local
180 xla::XlaOp loss, backprop; in Compile() local
/external/libopus/dnn/torch/lossgen/
Dtrain_lossgen.py75 loss = loss.to(device) variable
83loss = torch.mean(-target*torch.log(out+epsilon) - (1-target)*torch.log(1-out+epsilon)) variable
Dlossgen.py16 def forward(self, loss, perc, states=None): argument
Dtest_lossgen.py38 loss = one if np.random.rand() < prob else zero variable
/external/pytorch/test/cpp/jit/
Dtest_lite_trainer.cpp55 auto loss = ::torch::l1_loss(output, targets); in TEST() local
73 auto loss = ::torch::l1_loss(output, targets); in TEST() local
285 auto loss = ::torch::l1_loss(output, targets); in TEST() local
304 auto loss = ::torch::l1_loss(output, targets); in TEST() local
/external/pytorch/aten/src/ATen/native/
DLoss.cpp115 Tensor loss; in TORCH_IMPL_FUNC() local
131 Tensor loss; in TORCH_IMPL_FUNC() local
258 Tensor loss = at::empty_like(input); in binary_cross_entropy_cpu() local
263 …, const Tensor& target, const std::optional<Tensor>& weight_opt, int64_t reduction, Tensor& loss) { in binary_cross_entropy_out_cpu()
376 Tensor loss = (1 - target).mul_(input).sub_(log_sigmoid_input); in binary_cross_entropy_with_logits() local
387 Tensor loss; in poisson_nll_loss() local
/external/libopus/dnn/torch/neural-pitch/
Dtraining.py109loss = loss_custom(logits = pi,labels = yi,confidence = ci,choice = args.choice_cel,nmax = args.ou… variable
130loss = loss_custom(logits = pi,labels = yi,confidence = ci,choice = args.choice_cel,nmax = args.ou… variable
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_gradient.py28 def get_gradients_through_compute_gradients(optimizer, loss, activations): argument
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py461 def _transform_loss(self, loss): argument
465 def _get_gradients(self, tape, loss, var_list, grad_loss=None): argument
502 def minimize(self, loss, var_list, grad_loss=None, name=None, tape=None): argument
537 def _compute_gradients(self, loss, var_list, grad_loss=None, tape=None): argument
742 def get_gradients(self, loss, params): argument
771 def get_updates(self, loss, params): argument
/external/pytorch/torch/csrc/api/include/torch/nn/modules/
Dadaptive.h23 double loss; member
/external/executorch/examples/models/phi-3-mini-lora/
Dexport_model.py26 def __init__(self, model, loss): argument
/external/tensorflow/tensorflow/python/framework/experimental/
Dnn_ops.cc49 AbstractTensorHandle* loss; in PYBIND11_MODULE() local
/external/pytorch/scripts/release_notes/
Dclassifier.py266 def train_step(batch, model, optimizer, loss): argument
277 def eval_step(batch, model, loss): argument
/external/pytorch/functorch/examples/lennard_jones/
Dlennard_jones.py71 loss = loss_fn(training_energies, training_forces, energies, forces) variable
/external/tensorflow/tensorflow/python/debug/examples/v2/
Ddebug_mnist_v2.py200 def loss(probs, labels): function
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_output.py267 def __init__(self, loss=None, predictions=None, metrics=None): argument
381 def loss(self): member in _SupervisedOutput

12345