Home
last modified time | relevance | path

Searched refs:IsTensorRTCandidate (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc643 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() argument
711 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 …]
Dconvert_nodes.h398 Status IsTensorRTCandidate(const Node* node);
Dconvert_graph.cc746 std::bind(&TrtNodeValidator::IsTensorRTCandidate, &validator, in ConvertAfterShapes()
Dconvert_nodes.cc1188 Status TrtNodeValidator::IsTensorRTCandidate(const Node* node) { in IsTensorRTCandidate() function in tensorflow::tensorrt::convert::TrtNodeValidator