Searched refs:eval_config (Results 1 – 5 of 5) sorted by relevance
77 EvaluationStageConfig eval_config; in Init() local78 eval_config.set_name("average_precision"); in Init()79 *eval_config.mutable_specification() in Init()82 eval_config.mutable_specification() in Init()85 eval_stage_.reset(new ObjectDetectionAveragePrecisionStage(eval_config)); in Init()
81 EvaluationStageConfig eval_config; in RunImpl() local82 eval_config.set_name("inference_profiling"); in RunImpl()84 eval_config.mutable_specification()->mutable_tflite_inference_params(); in RunImpl()97 InferenceProfilerStage eval(eval_config); in RunImpl()
148 EvaluationStageConfig eval_config; in RunImpl() local149 eval_config.set_name("image_classification"); in RunImpl()150 auto* classification_params = eval_config.mutable_specification() in RunImpl()158 ImageClassificationStage eval(eval_config); in RunImpl()
125 EvaluationStageConfig eval_config; in RunImpl() local126 eval_config.set_name("object_detection"); in RunImpl()128 eval_config.mutable_specification()->mutable_object_detection_params(); in RunImpl()140 ObjectDetectionStage eval(eval_config); in RunImpl()
95 eval_config=None, argument150 config=eval_config,185 eval_config = config_pb2.ConfigProto()186 eval_config.experimental.collective_group_leader = ''206 eval_config=eval_config,