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