Home
last modified time | relevance | path

Searched defs:num_classes (Results 1 – 25 of 76) sorted by relevance

1234

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dstat_utils.cc33 float GiniImpurity(const LeafStat& stats, int32 num_classes) { in GiniImpurity()
40 float WeightedGiniImpurity(const LeafStat& stats, int32 num_classes) { in WeightedGiniImpurity()
74 float SmoothedGini(float sum, float square, int num_classes) { in SmoothedGini()
80 float WeightedSmoothedGini(float sum, float square, int num_classes) { in WeightedSmoothedGini()
/external/tensorflow/tensorflow/contrib/slim/python/slim/nets/
Dresnet_v2.py132 num_classes=None, argument
265 num_classes=None, argument
291 num_classes=None, argument
317 num_classes=None, argument
343 num_classes=None, argument
Dresnet_v1.py130 num_classes=None, argument
252 num_classes=None, argument
278 num_classes=None, argument
304 num_classes=None, argument
330 num_classes=None, argument
Dvgg.py75 num_classes=1000, argument
140 num_classes=1000, argument
205 num_classes=1000, argument
Doverfeat.py60 num_classes=1000, argument
Dalexnet.py64 num_classes=1000, argument
Dinception_v1.py305 num_classes=1000, argument
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_decoder.h43 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/
Dctc_decoder.h43 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()
Dctc_loss_calculator.h123 auto num_classes = inputs[0].cols(); in CalculateLoss() local
280 int batch_size, int num_classes, const Vector& seq_len, in PopulateLPrimes()
Dctc_beam_search_test.cc107 const int num_classes = 6; in TEST() local
187 const int num_classes = 6; in TEST() local
257 const int num_classes = 6; in TEST() local
/external/tensorflow/tensorflow/python/keras/
Dtesting_utils.py43 num_classes, argument
288 def get_small_sequential_mlp(num_hidden, num_classes, input_dim=None): argument
300 def get_small_functional_mlp(num_hidden, num_classes, input_dim): argument
311 def __init__(self, num_hidden, num_classes): argument
325 def __init__(self, num_hidden, num_classes): argument
342 def get_small_subclass_mlp(num_hidden, num_classes): argument
346 def get_small_subclass_mlp_with_custom_build(num_hidden, num_classes): argument
350 def get_small_mlp(num_hidden, num_classes, input_dim): argument
/external/tensorflow/tensorflow/python/ops/
Dconfusion_matrix.py96 num_classes=None, argument
206 num_classes=None, argument
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils_test.cc98 const int32 num_classes = 4; in TEST() local
116 const int32 num_classes = 4; in TEST() local
135 const int32 num_classes = 4; in TEST() local
169 const int32 num_classes = 4; in TEST() local
Dtree_utils.cc68 const Eigen::Tensor<float, 1, Eigen::RowMajor>& rights, int32 num_classes, in ClassificationSplitScore()
86 const int32 num_classes = in GetTwoBestClassification() local
238 const int32 num_classes = in MakeBootstrapWeights() local
284 const int32 num_classes = in BestSplitDominatesClassificationBootstrap() local
328 const int32 num_classes = in BestSplitDominatesClassificationHoeffding() local
349 const int32 num_classes = in DirichletCovarianceTrace() local
376 const int32 num_classes = in getDirichletMean() local
511 const int32 num_classes = in BestSplitDominatesClassificationChebyshev() local
/external/tensorflow/tensorflow/contrib/nn/python/ops/
Dsampling_ops.py116 num_classes, argument
252 num_classes, argument
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dconfusion_matrix_ops.py25 def confusion_matrix(labels, predictions, num_classes=None, dtype=dtypes.int32, argument
/external/tensorflow/tensorflow/python/keras/utils/
Dnp_utils.py25 def to_categorical(y, num_classes=None, dtype='float32'): argument
/external/tensorflow/tensorflow/core/kernels/
Dxent_op_test.cc24 static Graph* Xent(int batch_size, int num_classes) { in Xent()
Dsoftmax_op_functor.h50 const int num_classes = logits.dimension(kClassDim); in Compute() local
Dsparse_xent_op_test.cc26 static Graph* SparseXent(int batch_size, int num_classes) { in SparseXent()
Dxent_op.h70 const int num_classes = shape[kClassDim]; in Compute() local
Dmultinomial_op_test.cc27 static Graph* Multinomial(int batch_size, int num_classes, int num_samples) { in Multinomial()
/external/tensorflow/tensorflow/contrib/learn/python/learn/datasets/
Dmnist.py76 def dense_to_one_hot(labels_dense, num_classes): argument
86 def extract_labels(f, one_hot=False, num_classes=10): argument
/external/tensorflow/tensorflow/python/grappler/
Dgraph_placer_test.py39 num_classes=1024, argument

1234