Searched defs:max_total_size (Results 1 – 6 of 6) sorted by relevance
33 Tensor max_total_size(9000); in CombinedNonMaxSuppression() local
933 const Tensor& max_total_size = context->input(3); in Compute() local
204 ShapeHandle max_total_size; in CombinedNMSShapeFn() local
4956 max_total_size, argument
3139 auto max_total_size = in TEST_P() local3163 const int32 max_total_size; in TEST_P() member
5972 int max_total_size = *(static_cast<int*>(total_size.GetValues())); in ConvertCombinedNMS() local