Searched defs:num_output (Results 1 – 10 of 10) sorted by relevance
76 int num_output = num_cell; in Run() local
68 int num_output = in Run() local
16 const auto num_output = n->outputs().size(); in LayoutPropagation() local
593 const auto num_output = n->is(attr::output_layouts).size(); in setOpaqueLayout() local607 const auto num_output = n->is(attr::output_layouts).size(); in useOpaqueLayout() local
183 …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() local426 … const auto num_output = std::min<unsigned int>(_info.max_detections(), selected_indices.size()); in run() local
125 const unsigned int num_output = std::min(_max_output_size, num_above_thd); in run() local
19 void av1_nn_output_prec_reduce(float *const output, int num_output) { in av1_nn_output_prec_reduce()
504 const int num_output = op_kernel_context->num_outputs(); in SplitInputTask() local
52 const int num_output = output.size(0) * output.stride(0); in conv_depthwise3d_cuda_kernel() local
2101 int num_output; in TEST_P() member