Home
last modified time | relevance | path

Searched defs:n_classes (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DLoss.cu178 int64_t n_classes, in nll_loss_forward_no_reduce_cuda_kernel()
201 int64_t n_classes, in nll_loss_forward_reduce_cuda_kernel_1d()
239 int64_t n_classes, in nll_loss_forward_reduce_cuda_kernel_2d()
287 int64_t n_classes = input.size(-1); in nll_loss_forward_out_cuda_template() local
414 int64_t n_classes, in nll_loss_backward_no_reduce_cuda_kernel()
436 int64_t n_classes, in nll_loss_backward_reduce_cuda_kernel_1d()
461 int64_t n_classes, in nll_loss_backward_reduce_cuda_kernel_2d()
496 int64_t n_classes = input.size(-1); in nll_loss_backward_out_cuda_template() local
H A DNLLLoss2d.cu59 int64_t n_classes = input.size(1); in C10_LAUNCH_BOUNDS_1() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLossNLL.cpp167 const auto n_classes = input.size(-1); in nll_loss_out_frame() local
347 const auto n_classes = input.size(-1); in nll_loss_backward_out_frame() local
497 const auto n_classes = self.size(class_dim); in cross_entropy_loss_prob_target() local
569 auto n_classes = input.sym_size(class_dim); in cross_entropy_loss_label_smoothing() local
H A DLossNLL2d.cpp110 const int64_t n_classes = input.size(1); in nll_loss2d_forward_out_frame() local
342 const int64_t n_classes = input.size(1); in nll_loss2d_backward_out_frame() local
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dadaptive.cpp8 int64_t n_classes, in AdaptiveLogSoftmaxWithLossOptions()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dadaptive.h49 int64_t n_classes, in AdaptiveLogSoftmaxWithLossImpl()
/aosp_15_r20/external/pytorch/functorch/examples/ensembling/
H A Dparallel_train.py62 def __init__(self, hidden_dim=32, n_classes=2): argument
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_eager_transforms.py841 def __init__(self, hidden_dim=32, n_classes=2): argument
863 def __init__(self, hidden_dim=32, n_classes=2): argument
4378 def __init__(self, hidden_dim=32, n_classes=2): argument
4479 def __init__(self, hidden_dim=32, n_classes=2): argument