Searched defs:num_labels (Results 1 – 13 of 13) sorted by relevance
| /external/compiler-rt/test/dfsan/ |
| D | dump_labels.c | 39 unsigned long num_labels = 1 << (sizeof(dfsan_label) * 8); in main() local
|
| /external/tflite-support/tensorflow_lite_support/examples/task/vision/desktop/ |
| D | image_segmenter_demo.cc | 109 const int num_labels = segmentation.colored_labels_size(); in DisplayColorLegend() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 596 def _ilabel_to_state(labels, num_labels, ilabel_log_probs): argument
612 def _state_to_olabel(labels, num_labels, states): argument
631 def _state_to_olabel_unique(labels, num_labels, states, unique): argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | einsum_op_util.cc | 105 int num_labels = label_mapping.size(); in ParseEinsumEquation() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator_test.cc | 54 const int num_labels = 10; in CreateMiniGraph() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | LossCTC.cu | 235 int64_t num_labels = log_probs.size(2); in ctc_loss_gpu_template() local
515 … int64_t batch_size, int64_t num_labels, int64_t BLANK, bool zero_infinity) { in ctc_loss_backward_collect_gpu_kernel()
584 int64_t num_labels /* D */ ) { in ctc_loss_zero_padded_gradients()
610 int64_t num_labels = log_probs.size(2); in ctc_loss_backward_gpu_template() local
|
| /external/mesa3d/src/freedreno/afuc/ |
| D | asm.c | 76 static unsigned num_labels; variable
|
| /external/ComputeLibrary/examples/ |
| D | neon_cnn.cpp | 117 constexpr unsigned int num_labels = 128; in do_setup() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_einsum_op.cc | 197 int num_labels = label_types.size(); in Compute() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | LossCTC.cpp | 68 int64_t num_labels = log_probs.size(2); in ctc_loss_allocate_outputs() local
239 int64_t num_labels = log_probs.size(2); in ctc_loss_backward_cpu_template() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 534 int num_labels = label_types.size(); in Compute() local
|
| /external/mesa3d/src/gallium/frontends/d3d10umd/ |
| D | ShaderTGSI.c | 242 uint num_labels; member
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics.py | 2090 num_labels=None, argument
|