| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | testing_utils.py | 60 num_classes, argument
422 def get_small_sequential_mlp(num_hidden, num_classes, input_dim=None): argument
433 def get_small_functional_mlp(num_hidden, num_classes, input_dim): argument
446 num_classes, argument
474 def __init__(self, num_hidden, num_classes): argument
491 def get_small_subclass_mlp(num_hidden, num_classes): argument
495 def get_small_subclass_mlp_with_custom_build(num_hidden, num_classes): argument
499 def get_small_mlp(num_hidden, num_classes, input_dim): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/ |
| H A D | torchvision_models.py | 150 num_classes=1000, argument
420 def _segm_resnet(name, backbone_name, num_classes, aux, pretrained_backbone=True): argument
453 arch_type, backbone, pretrained, progress, num_classes, aux_loss, **kwargs argument
470 pretrained=False, progress=True, num_classes=21, aux_loss=None, **kwargs argument
499 num_classes, argument
668 def __init__(self, num_classes, matcher, weight_dict, eos_coef, losses): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ctc/ |
| H A D | ctc_decoder.h | 43 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder()
61 int num_classes() { return num_classes_; } in num_classes() function
74 CTCGreedyDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCGreedyDecoder()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ctc/ |
| H A D | ctc_decoder.h | 46 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder()
64 int num_classes() { return num_classes_; } in num_classes() function
79 CTCGreedyDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCGreedyDecoder()
|
| H A D | ctc_beam_search_test.cc | 109 const int num_classes = 6; in ctc_beam_search_decoding_with_and_without_dictionary() local
199 const int num_classes = 6; in ctc_beam_search_decoding_all_beam_elements_have_finite_scores() local
277 const int num_classes = 6; in ctc_beam_search_label_selection() local
|
| H A D | ctc_loss_calculator.h | 131 auto num_classes = inputs[0].cols(); in CalculateLoss() local
289 int batch_size, int num_classes, const Vector& seq_len, in PopulateLPrimes()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_expanded_weights.py | 478 def convnet(num_classes, num_dim): argument
501 def convnet(num_classes, num_dim): argument
527 def instance_norm_model(num_classes, num_dim): argument
553 def group_norm_model(num_classes, num_dim): argument
572 def layer_norm_model(num_classes, num_dim): argument
590 def embedding_model(num_classes, num_embedding): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_impl.py | 1830 num_classes, argument
2017 num_classes, argument
2122 num_classes, argument
2234 num_classes, argument
2326 num_classes, argument
|
| H A D | confusion_matrix.py | 94 num_classes=None, argument
202 num_classes=None, argument
|
| H A D | metrics_impl.py | 228 def _streaming_confusion_matrix(labels, predictions, num_classes, weights=None): argument
1177 num_classes, argument
1282 num_classes, argument
3365 def _clean_out_of_range_indices(labels, num_classes): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | detection_postprocess.cc | 95 int num_classes; member
507 int num_classes; member
637 const int num_classes = op_data->num_classes; in NonMaxSuppressionMultiClassRegularHelper() local
776 const int num_classes = op_data->num_classes; in NonMaxSuppressionMultiClassFastHelper() local
858 const int num_classes = op_data->num_classes; in NonMaxSuppressionMultiClass() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | in_topk_op_gpu.cu.cc | 43 int num_targets, int num_classes) { in ComputePredictionMaskKernel()
96 const Eigen::Index num_classes = predictions.dimension(1); in operator ()() local
|
| H A D | ctc_loss_op.cc | 122 const int num_classes = static_cast<const int>(num_classes_raw); in Compute() local
298 const int num_classes = static_cast<const int>(num_classes_raw); in Compute() local
|
| H A D | ctc_decoder_ops.cc | 213 const int num_classes = static_cast<const int>(num_classes_raw); in Compute() local
321 const int num_classes = static_cast<const int>(num_classes_raw); in Compute() local
|
| H A D | xent_op.h | 70 const int num_classes = shape[kClassDim]; in Compute() local
|
| H A D | softmax_op_functor.h | 50 const int num_classes = logits.dimension(kClassDim); in Compute() local
|
| H A D | xent_op_test.cc | 25 static Graph* Xent(int batch_size, int num_classes, DataType type) { in Xent()
|
| H A D | multinomial_op_test.cc | 27 static Graph* Multinomial(int batch_size, int num_classes, int num_samples) { in Multinomial()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/boundingboxtransform/generic/neon/ |
| H A D | impl.cpp | 32 const size_t num_classes = deltas->info()->tensor_shape()[0] >> 2; in bounding_box_transform_qsymm16() local
89 const size_t num_classes = deltas->info()->tensor_shape()[0] >> 2; in bounding_box_transform() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | non_max_suppression_op.cc | 98 const int num_classes) { in ParseAndCheckCombinedNMSBoxSizes()
341 int num_classes, const int size_per_class, in DoNMSPerClass()
470 int num_classes = inp_scores.dim_size(2); in BatchedNonMaxSuppressionOp() local
1017 const int num_classes = scores.dim_size(2); in Compute() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CPP/kernels/ |
| H A D | CPPBoxWithNonMaximaSuppressionLimitKernel.cpp | 201 const int num_classes = _scores_in->info()->dimension(0); in run_nmslimit() local
352 const unsigned int num_classes = scores_in->info()->dimension(0); in configure() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_ssd_mobilenet.cpp | 369 const int num_classes = 21; in create_graph_float() local
672 const int num_classes = 90; in create_graph_qasymm() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/ |
| H A D | np_utils.py | 22 def to_categorical(y, num_classes=None, dtype='float32'): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Onehot.cpp | 17 Tensor one_hot(const Tensor &self, int64_t num_classes) { in one_hot()
|
| /aosp_15_r20/external/ComputeLibrary/utils/ |
| H A D | GraphUtils.cpp | 466 const size_t num_classes = tensor.info()->dimension(0); in access_predictions_tensor() local
608 const size_t num_classes = tensor.info()->dimension(0); in access_predictions_tensor() local
|