| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Loss.cu | 178 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
|
| D | NLLLoss2d.cu | 59 int64_t n_classes = input.size(1); in C10_LAUNCH_BOUNDS_1() local
|
| /external/pytorch/torch/csrc/api/src/nn/options/ |
| D | adaptive.cpp | 8 int64_t n_classes, in AdaptiveLogSoftmaxWithLossOptions()
|
| /external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| D | adaptive.h | 49 int64_t n_classes, in AdaptiveLogSoftmaxWithLossImpl()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | LossNLL.cpp | 167 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
|
| D | LossNLL2d.cpp | 110 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
|
| /external/pytorch/functorch/examples/ensembling/ |
| D | parallel_train.py | 62 def __init__(self, hidden_dim=32, n_classes=2): argument
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/read-fonts-0.23.2/src/tables/ |
| D | aat.rs | 319 let n_classes = self.header.state_size() as usize; in entry() localVariable
393 let n_classes = self.header.n_classes() as usize; in entry() localVariable
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/read-fonts-0.23.2/src/tables/ |
| D | aat.rs | 319 let n_classes = self.header.state_size() as usize; in entry() localVariable
393 let n_classes = self.header.n_classes() as usize; in entry() localVariable
|
| /external/pytorch/test/functorch/ |
| D | test_eager_transforms.py | 841 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
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/read-fonts-0.23.2/generated/ |
| D | generated_aat.rs | 1137 pub fn n_classes(&self) -> u32 { in n_classes() method
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/read-fonts-0.23.2/generated/ |
| D | generated_aat.rs | 1137 pub fn n_classes(&self) -> u32 { in n_classes() method
|