Searched refs:ConvertToTensorOrWeights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 413 Status ConvertToTensorOrWeights(const NodeDef& node_def, int output_port,
|
| D | convert_nodes_test.cc | 534 Status ConvertToTensorOrWeights(const Scope& scope, const Node* node, in ConvertToTensorOrWeights() function in tensorflow::tensorrt::convert::ValidatorTest
545 return validator.ConvertToTensorOrWeights(node->def(), output_port, in ConvertToTensorOrWeights()
565 TEST_F(ValidatorTest, ConvertToTensorOrWeights) { in TEST_F() argument
572 ExpectStatus(ConvertToTensorOrWeights(s, node.op().node(), in TEST_F()
584 return this->ConvertToTensorOrWeights(s, add.operation.node(), in TEST_F()
|
| D | convert_nodes.cc | 1147 Status TrtNodeValidator::ConvertToTensorOrWeights( in ConvertToTensorOrWeights() function in tensorflow::tensorrt::convert::TrtNodeValidator
1211 Status status = ConvertToTensorOrWeights(src_def, edge->src_output(), in IsTensorRTCandidate()
|