Searched refs:IsTensorRTCandidate (Results 1 – 4 of 4) sorted by relevance
643 ExpectStatus(validator.IsTensorRTCandidate(add_node), error::UNIMPLEMENTED, in TEST_F()648 TF_EXPECT_OK(validator.IsTensorRTCandidate(add_node)); in TEST_F()653 ExpectStatus(validator.IsTensorRTCandidate(add_node), in TEST_F()657 TEST(TrtNodeValidator, IsTensorRTCandidate) { in TEST() argument711 TF_EXPECT_OK(validator.IsTensorRTCandidate(matmul.operation.node())); in TEST()713 validator.IsTensorRTCandidate(incompatible_matmul.operation.node()), in TEST()717 ExpectStatus(validator.IsTensorRTCandidate(unsupported_op.operation.node()), in TEST()719 ExpectStatus(validator.IsTensorRTCandidate( in TEST()724 TF_EXPECT_OK(validator.IsTensorRTCandidate(quantize.operation.node())); in TEST()726 ExpectStatus(validator.IsTensorRTCandidate(quantize.operation.node()), in TEST()[all …]
398 Status IsTensorRTCandidate(const Node* node);
746 std::bind(&TrtNodeValidator::IsTensorRTCandidate, &validator, in ConvertAfterShapes()
1188 Status TrtNodeValidator::IsTensorRTCandidate(const Node* node) { in IsTensorRTCandidate() function in tensorflow::tensorrt::convert::TrtNodeValidator