Searched refs:TrtCandidateSelector (Results 1 – 3 of 3) sorted by relevance
36 class TrtCandidateSelector {38 TrtCandidateSelector(const grappler::GraphProperties& graph_properties,
52 TEST(TrtCandidateSelector, Basics) { in TEST() argument103 TrtCandidateSelector selector(graph_properties, precision_mode); in TEST()
68 TrtCandidateSelector::TrtCandidateSelector( in TrtCandidateSelector() function in tensorflow::tensorrt::convert::TrtCandidateSelector73 Status TrtCandidateSelector::IsTensorRTCandidate(const Node* node) { in IsTensorRTCandidate()817 TrtCandidateSelector candidate_selector(*params.graph_properties, in ConvertAfterShapes()821 std::bind(&TrtCandidateSelector::IsTensorRTCandidate, &candidate_selector, in ConvertAfterShapes()