Home
last modified time | relevance | path

Searched defs:num_output (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/CPP/functions/
H A DCPPDetectionPostProcessLayer.cpp183 …std::vector<unsigned int> &sorted_indices, const unsigned int num_output, const unsigned int max_d… in SaveOutputs()
358 const auto num_output = std::min<unsigned int>(max_detections, num_selected); in run() local
426 … const auto num_output = std::min<unsigned int>(_info.max_detections(), selected_indices.size()); in run() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Dgraph_helper.cpp593 const auto num_output = n->is(attr::output_layouts).size(); in setOpaqueLayout() local
607 const auto num_output = n->is(attr::output_layouts).size(); in useOpaqueLayout() local
H A Dlayout_propagation.cpp16 const auto num_output = n->outputs().size(); in LayoutPropagation() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Didentify_lstm_split_inputs.cc76 int num_output = num_cell; in Run() local
H A Didentify_lstm_merge_inputs.cc68 int num_output = in Run() local
/aosp_15_r20/external/ComputeLibrary/src/core/CPP/kernels/
H A DCPPNonMaximumSuppressionKernel.cpp125 const unsigned int num_output = std::min(_max_output_size, num_above_thd); in run() local
/aosp_15_r20/external/libaom/av1/encoder/
H A Dml.c19 void av1_nn_output_prec_reduce(float *const output, int num_output) { in av1_nn_output_prec_reduce()
/aosp_15_r20/frameworks/base/media/jni/
H A DJetPlayer.cpp203 int num_output = 0; in render() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDepthwiseConv3d.cu52 const int num_output = output.size(0) * output.stride(0); in conv_depthwise3d_cuda_kernel() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbatch_resource_base.cc504 const int num_output = op_kernel_context->num_outputs(); in SplitInputTask() local
/aosp_15_r20/external/webrtc/modules/audio_processing/
H A Daudio_processing_unittest.cc2101 int num_output; in TEST_P() member