revert-13637-optimize-opyreader
nhzlx 7 years ago
parent f277f53c33
commit 1f6c9dbad4

@ -90,7 +90,7 @@ class TensorRTSubgraphPredictor : public NativePaddlePredictor {
// Analyze inference_program
Argument argument;
int* minimum_subgraph_size = new int(config_.minimun_subgraph_size);
int* minimum_subgraph_size = new int(config_.minimum_subgraph_size);
int* max_batch_size = new int(config_.max_batch_size);
int* workspace_size = new int(config_.workspace_size);
std::string* precision_mode = new std::string(config_.precision_mode);

@ -155,7 +155,7 @@ struct TensorRTConfig : public NativeConfig {
// We set this variable to control the minimum number of nodes in the
// subgraph, 3 as
// default value.
int minimun_subgraph_size = 3;
int minimum_subgraph_size = 3;
// Reserved configuration
// We just support "FP32" now, "FP16" and "INT8" will be supported.
std::string precision_mode = "FP32";

Loading…
Cancel
Save