| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_subclassing_test_util.py | 28 def __init__(self, num_classes=10): argument
42 def get_multi_io_subclass_model(use_bn=False, use_dp=False, num_classes=(2, 3)): argument
65 def __init__(self, num_classes=2): argument
85 def __init__(self, num_classes=2): argument
94 def get_functional_graph_model(input_dim, num_classes): argument
109 def get_nested_model_3(input_dim, num_classes): argument
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| 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()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| 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()
|
| 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
|
| 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()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | confusion_matrix.py | 98 num_classes=None, argument
208 num_classes=None, argument
|
| D | nn_impl.py | 1621 num_classes, argument
1807 num_classes, argument
1911 num_classes, argument
2022 num_classes, argument
2113 num_classes, argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 48 num_classes, argument
399 def get_small_sequential_mlp(num_hidden, num_classes, input_dim=None): argument
411 def get_small_functional_mlp(num_hidden, num_classes, input_dim): argument
422 def __init__(self, num_hidden, num_classes, use_bn=False, use_dp=False): argument
447 def __init__(self, num_hidden, num_classes): argument
464 def get_small_subclass_mlp(num_hidden, num_classes): argument
468 def get_small_subclass_mlp_with_custom_build(num_hidden, num_classes): argument
472 def get_small_mlp(num_hidden, num_classes, input_dim): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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
|
| D | xent_op_test.cc | 24 static Graph* Xent(int batch_size, int num_classes) { in Xent()
|
| D | softmax_op_functor.h | 50 const int num_classes = logits.dimension(kClassDim); in Compute() local
|
| D | sparse_xent_op_test.cc | 26 static Graph* SparseXent(int batch_size, int num_classes) { in SparseXent()
|
| D | xent_op.h | 70 const int num_classes = shape[kClassDim]; in Compute() local
|
| D | multinomial_op_test.cc | 27 static Graph* Multinomial(int batch_size, int num_classes, int num_samples) { in Multinomial()
|
| D | one_hot_op_test.cc | 25 static Graph* OneHot(int batch_size, int num_classes, int axis) { in OneHot()
|
| D | in_topk_op_test.cc | 36 static Graph* InTopK(int num_targets, int num_classes, T top_k) { in InTopK()
|
| D | ctc_loss_op.cc | 110 const int num_classes = static_cast<const int>(num_classes_raw); in Compute() local
286 const int num_classes = static_cast<const int>(num_classes_raw); in Compute() local
|
| D | multinomial_op_gpu.cu.cc | 47 __global__ void MultinomialKernel(int32 nthreads, const int32 num_classes, in MultinomialKernel()
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | np_utils.py | 25 def to_categorical(y, num_classes=None, dtype='float32'): argument
|
| /external/tensorflow/tensorflow/examples/tutorials/mnist/ |
| D | input_data.py | 78 def _dense_to_one_hot(labels_dense, num_classes): argument
88 def _extract_labels(f, one_hot=False, num_classes=10): argument
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | graph_placer_test.py | 39 num_classes=1024, argument
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_image_model_correctness_test.py | 68 def _get_data(self, count, shape=(28, 28, 3), num_classes=10): argument
|
| D | keras_embedding_model_correctness_test.py | 132 num_classes=2): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | categorical_op.cc | 65 const int64 num_classes = logits_shape.dim_size(1); in Compile() local
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
| D | object_detection_average_precision_stage_test.cc | 32 EvaluationStageConfig GetAveragePrecisionStageConfig(int num_classes) { in GetAveragePrecisionStageConfig()
|