Searched refs:TrtNodeValidator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 387 class TrtNodeValidator {
392 TrtNodeValidator(const grappler::GraphProperties& graph_properties,
|
| D | convert_graph.cc | 742 TrtNodeValidator validator(static_graph_properties, params.precision_mode, in ConvertAfterShapes()
746 std::bind(&TrtNodeValidator::IsTensorRTCandidate, &validator, in ConvertAfterShapes()
|
| D | convert_nodes_test.cc | 530 TrtNodeValidator* validator) { in op_validators()
542 TrtNodeValidator validator(graph_properties, TrtPrecisionMode::FP32, in ConvertToTensorOrWeights()
549 const std::set<string>* GetQuantizeOps(TrtNodeValidator* validator) { in GetQuantizeOps()
557 TrtNodeValidator validator(graph_properties, TrtPrecisionMode::FP32, in TEST_F()
628 TrtNodeValidator validator(graph_properties, TrtPrecisionMode::FP32, in TEST_F()
657 TEST(TrtNodeValidator, IsTensorRTCandidate) { in TEST() argument
708 TrtNodeValidator validator(graph_properties, precision_mode, in TEST()
1622 TrtNodeValidator validator(graph_properties, converter_->precision_mode(), in RunValidation()
|
| D | convert_nodes.cc | 1129 const std::set<string>* TrtNodeValidator::quantize_ops = new std::set<string>{
1136 TrtNodeValidator::TrtNodeValidator( in TrtNodeValidator() function in tensorflow::tensorrt::convert::TrtNodeValidator
1147 Status TrtNodeValidator::ConvertToTensorOrWeights( in ConvertToTensorOrWeights()
1188 Status TrtNodeValidator::IsTensorRTCandidate(const Node* node) { in IsTensorRTCandidate()
1228 Status TrtNodeValidator::ConvertConstToWeights( in ConvertConstToWeights()
6311 void TrtNodeValidator::RegisterOpValidators() { in RegisterOpValidators()
|