| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | model_subclassing_test_util.py | 24 def __init__(self, num_classes=10): argument
38 def get_multi_io_subclass_model(use_bn=False, use_dp=False, num_classes=(2, 3)): argument
61 def __init__(self, num_classes=2): argument
81 def __init__(self, num_classes=2): argument
90 def get_functional_graph_model(input_dim, num_classes): argument
105 def get_nested_model_3(input_dim, num_classes): argument
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | YOLOLayerFixture.h | 50 …ctivationLayerInfo::ActivationFunction function, float alpha_beta, int32_t num_classes, DataLayout… in setup()
73 …TensorShape shape, bool in_place, const ActivationLayerInfo &info, int32_t num_classes, DataLayout… in compute_target()
120 …ference(const TensorShape &shape, const ActivationLayerInfo &info, int32_t num_classes, DataType d… in compute_reference()
142 …ctivationLayerInfo::ActivationFunction function, float alpha_beta, int32_t num_classes, DataLayout… in setup()
153 …ctivationLayerInfo::ActivationFunction function, float alpha_beta, int32_t num_classes, DataLayout… in setup()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLYOLOLayer.cpp | 32 …gure(ICLTensor *input, ICLTensor *output, const ActivationLayerInfo &act_info, int32_t num_classes) in configure()
37 …ext, ICLTensor *input, ICLTensor *output, const ActivationLayerInfo &act_info, int32_t num_classes) in configure()
44 …orInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info, int32_t num_classes) in validate()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLYOLOLayerKernel.cpp | 46 …orInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info, int32_t num_classes) in validate_arguments()
105 …gure(ICLTensor *input, ICLTensor *output, const ActivationLayerInfo &act_info, int32_t num_classes) in configure()
110 …ext, ICLTensor *input, ICLTensor *output, const ActivationLayerInfo &act_info, int32_t num_classes) in configure()
158 …orInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info, int32_t num_classes) in validate()
|
| /external/tensorflow/tensorflow/lite/kernels/ctc/ |
| 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
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEYOLOLayer.cpp | 31 …onfigure(ITensor *input, ITensor *output, const ActivationLayerInfo &act_info, int32_t num_classes) in configure()
38 …orInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info, int32_t num_classes) in validate()
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | YOLOLayerNode.cpp | 34 YOLOLayerNode::YOLOLayerNode(ActivationLayerInfo act_info, int32_t num_classes) in YOLOLayerNode()
46 int32_t YOLOLayerNode::num_classes() const in num_classes() function in arm_compute::graph::YOLOLayerNode
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | confusion_matrix.py | 98 num_classes=None, argument
206 num_classes=None, argument
|
| D | nn_impl.py | 1807 num_classes, argument
1994 num_classes, argument
2099 num_classes, argument
2211 num_classes, argument
2303 num_classes, argument
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | YOLOLayer.cpp | 40 …sor<T> yolo_layer(const SimpleTensor<T> &src, const ActivationLayerInfo &info, int32_t num_classes) in yolo_layer()
71 …er<uint8_t>(const SimpleTensor<uint8_t> &src, const ActivationLayerInfo &info, int32_t num_classes) in yolo_layer()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEYOLOLayerKernel.cpp | 47 …orInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info, int32_t num_classes) in validate_arguments()
152 …onfigure(ITensor *input, ITensor *output, const ActivationLayerInfo &act_info, int32_t num_classes) in configure()
194 …orInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info, int32_t num_classes) in validate()
|
| /external/tensorflow/tensorflow/python/keras/ |
| 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
|
| /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()
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | revive_test.py | 149 def __init__(self, num_classes, name=None): argument
158 def __init__(self, num_classes, name=None): argument
172 def __init__(self, num_classes, name=None): argument
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | np_utils.py | 22 def to_categorical(y, num_classes=None, dtype='float32'): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local_test.py | 533 num_classes, argument
567 num_classes, data_format): argument
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/integration_test/benchmarks/ |
| D | numpy_mlp.py | 34 def __init__(self, num_classes=NUM_CLASSES, input_size=INPUT_SIZE, argument
|