Home
last modified time | relevance | path

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

123

/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_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
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
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/core/util/ctc/
Dctc_decoder.h42 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder()
60 int num_classes() { return num_classes_; } in num_classes() function
73 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/_impl/keras/
Dmodel_subclassing_test.py41 def __init__(self, use_bn=False, use_dp=False, num_classes=10): argument
65 def __init__(self, use_bn=False, use_dp=False, num_classes=(2, 3)): argument
94 def __init__(self, num_classes=2): argument
111 def get_functional_graph_model(input_dim, num_classes): argument
124 def __init__(self, num_classes=2): argument
139 def get_nested_model_3(input_dim, num_classes): argument
Dtesting_utils.py31 num_classes): 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/_impl/keras/utils/
Dnp_utils.py25 def to_categorical(y, num_classes=None): 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.h61 const int num_classes = logits.dimension(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/python/ops/
Dconfusion_matrix.py98 def confusion_matrix(labels, predictions, num_classes=None, dtype=dtypes.int32, argument
/external/tensorflow/tensorflow/contrib/learn/python/learn/datasets/
Dmnist.py68 def dense_to_one_hot(labels_dense, num_classes): argument
77 def extract_labels(f, one_hot=False, num_classes=10): argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcategorical_op.cc57 const int64 num_classes = logits_shape.dim_size(1); in Compile() local

123