Home
last modified time | relevance | path

Searched refs:ConvertToTensorOrWeights (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h401 Status ConvertToTensorOrWeights(
Dconvert_nodes_test.cc401 Status ConvertToTensorOrWeights( in ConvertToTensorOrWeights() function in tensorflow::tensorrt::convert::ValidatorTest
405 return validator_.ConvertToTensorOrWeights( in ConvertToTensorOrWeights()
421 TEST_F(ValidatorTest, ConvertToTensorOrWeights) { in TEST_F() argument
428 ExpectStatus(ConvertToTensorOrWeights(node_def, /*output_port=*/0, in TEST_F()
446 return this->ConvertToTensorOrWeights(node_def, /*output_port=*/0, in TEST_F()
Dconvert_nodes.cc893 Status TrtNodeValidator::ConvertToTensorOrWeights( in ConvertToTensorOrWeights() function in tensorflow::tensorrt::convert::TrtNodeValidator
963 Status status = ConvertToTensorOrWeights( in ValidateNode()