Searched refs:StatSummarizer (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/util/ |
| D | stat_summarizer_wrapper.cc | 28 py::class_<tensorflow::StatSummarizer> stat_summ_class(m, "StatSummarizer", in PYBIND11_MODULE()
34 return new tensorflow::StatSummarizer(proto); in PYBIND11_MODULE()
37 return new tensorflow::StatSummarizer( in PYBIND11_MODULE()
40 .def("ProcessStepStats", &tensorflow::StatSummarizer::ProcessStepStats) in PYBIND11_MODULE()
41 .def("GetOutputString", &tensorflow::StatSummarizer::GetOutputString) in PYBIND11_MODULE()
42 .def("PrintStepStats", &tensorflow::StatSummarizer::PrintStepStats) in PYBIND11_MODULE()
43 .def("ProcessStepStatsStr", [](tensorflow::StatSummarizer& self, in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/tools/android/inference_interface/jni/ |
| D | run_stats_jni.cc | 26 using tensorflow::StatSummarizer;
29 StatSummarizer* requireHandle(JNIEnv* env, jlong handle) { in requireHandle()
35 return reinterpret_cast<StatSummarizer*>(handle); in requireHandle()
43 static_assert(sizeof(jlong) >= sizeof(StatSummarizer*), in RUN_STATS_METHOD()
46 return reinterpret_cast<jlong>(new StatSummarizer(opts)); in RUN_STATS_METHOD()
52 delete reinterpret_cast<StatSummarizer*>(handle); in RUN_STATS_METHOD()
57 StatSummarizer* s = requireHandle(env, handle); in RUN_STATS_METHOD()
74 StatSummarizer* s = requireHandle(env, handle); in RUN_STATS_METHOD()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | stat_summarizer.cc | 36 StatSummarizer::StatSummarizer(const StatSummarizerOptions& options) in StatSummarizer() function in tensorflow::StatSummarizer
39 StatSummarizer::StatSummarizer(const tensorflow::GraphDef& tensorflow_graph) in StatSummarizer() function in tensorflow::StatSummarizer
42 StatSummarizer::~StatSummarizer() {} in ~StatSummarizer()
44 void StatSummarizer::Validate(const std::vector<TensorDescription>* outputs, in Validate()
81 void StatSummarizer::PrintStepStats() const { in PrintStepStats()
130 void StatSummarizer::ProcessStepStats(const StepStats& step_stats) { in ProcessStepStats()
230 void StatSummarizer::PrintOutputs() const { in PrintOutputs()
|
| D | stat_summarizer.h | 47 class StatSummarizer {
49 explicit StatSummarizer(const StatSummarizerOptions& options);
53 explicit StatSummarizer(const tensorflow::GraphDef& tensorflow_graph);
55 ~StatSummarizer();
|
| D | stat_summarizer_test.cc | 73 StatSummarizer stats(graph_def); in TEST()
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model_test.cc | 68 std::unique_ptr<StatSummarizer> stats; in TEST()
69 stats.reset(new tensorflow::StatSummarizer(*(loaded_graph_def.get()))); in TEST()
93 std::unique_ptr<StatSummarizer> stats; in TEST()
94 stats.reset(new tensorflow::StatSummarizer(*(loaded_graph_def.get()))); in TEST()
|
| D | benchmark_model.h | 42 StatSummarizer* stats, int64* inference_time_us);
49 StatSummarizer* stats, int64* total_time_us,
|
| D | benchmark_model.cc | 287 StatSummarizer* stats, int64* inference_time_us) { in RunBenchmark()
325 StatSummarizer* stats, int64* total_time_us, in TimeMultipleRuns()
489 std::unique_ptr<StatSummarizer> stats; in Main()
522 stats.reset(new tensorflow::StatSummarizer(stats_options)); in Main()
|