Searched refs:eval_pipeline (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| D | eval_pipeline_builder_test.cc | 90 std::unique_ptr<EvalPipeline> eval_pipeline; in TEST() local
93 builder.WithInputStage(&input_stage).Build(scope, &eval_pipeline); in TEST()
95 EXPECT_FALSE(eval_pipeline); in TEST()
98 builder.WithRunModelStage(&run_model_stage).Build(scope, &eval_pipeline); in TEST()
100 EXPECT_FALSE(eval_pipeline); in TEST()
103 .Build(scope, &eval_pipeline); in TEST()
105 EXPECT_FALSE(eval_pipeline); in TEST()
108 builder.WithInput(pipeline_input, DT_FLOAT).Build(scope, &eval_pipeline); in TEST()
110 EXPECT_FALSE(eval_pipeline); in TEST()
112 status = builder.WithAccuracyEval(&eval).Build(scope, &eval_pipeline); in TEST()
[all …]
|
| D | eval_pipeline_test.cc | 67 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval); in TEST() local
69 TF_CHECK_OK(eval_pipeline.AttachSession(std::move(session))); in TEST()
70 TF_CHECK_OK(eval_pipeline.Run(CreateFloatTensor(5), CreateFloatTensor(27))); in TEST()
92 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval); in TEST() local
94 TF_CHECK_OK(eval_pipeline.AttachSession(std::move(session))); in TEST()
98 auto status = eval_pipeline.Run(string_tensor, CreateFloatTensor(27)); in TEST()
115 EvalPipeline eval_pipeline(graph_def, params, &accuracy_eval); in TEST() local
117 TF_CHECK_OK(eval_pipeline.AttachSession(std::move(session))); in TEST()
118 auto status = eval_pipeline.Run(CreateFloatTensor(5), CreateFloatTensor(27)); in TEST()
|
| D | README.md | 15 std::unique_ptr<EvalPipeline> eval_pipeline;
21 .Build(scope, &eval_pipeline);
26 TF_CHECK_OK(eval_pipeline.AttachSession(std::move(session)));
29 TF_CHECK_OK(eval_pipeline.Run(input1, ground_truth1));
|
| D | BUILD | 214 name = "eval_pipeline",
215 srcs = ["eval_pipeline.cc"],
216 hdrs = ["eval_pipeline.h"],
241 ":eval_pipeline",
266 ":eval_pipeline",
|
| D | eval_pipeline_builder.cc | 55 const Scope& scope, std::unique_ptr<EvalPipeline>* eval_pipeline) { in Build() argument
93 *eval_pipeline = in Build()
|
| D | eval_pipeline_builder.h | 86 std::unique_ptr<EvalPipeline>* eval_pipeline);
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/ |
| D | imagenet_model_evaluator.cc | 202 std::unique_ptr<EvalPipeline> eval_pipeline; in EvaluateModelForShard() local
209 .Build(root, &eval_pipeline); in EvaluateModelForShard()
214 TF_RETURN_IF_ERROR(eval_pipeline->AttachSession(std::move(session))); in EvaluateModelForShard()
217 TF_CHECK_OK(eval_pipeline->Run(CreateStringTensor(image_label.image), in EvaluateModelForShard()
|
| D | BUILD | 131 "//tensorflow/lite/tools/accuracy:eval_pipeline",
|