Home
last modified time | relevance | path

Searched refs:subgraph_index (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer.cc29 uint32_t subgraph_index; member
71 uint32_t subgraph_index, in GetOperatorDetails() argument
74 const_cast<tflite::Interpreter&>(interpreter).subgraph(subgraph_index); in GetOperatorDetails()
131 const auto subgraph_index = event->extra_event_metadata; in ProcessProfiles() local
132 auto stats_calculator = GetStatsCalculator(subgraph_index); in ProcessProfiles()
144 GetOperatorDetails(interpreter, subgraph_index, node_index); in ProcessProfiles()
187 total_us_per_subgraph_map[subgraph_index] += node_exec_time; in ProcessProfiles()
205 uint32_t subgraph_index) { in GetStatsCalculator() argument
206 if (stats_calculator_map_.count(subgraph_index) == 0) { in GetStatsCalculator()
207 stats_calculator_map_[subgraph_index] = in GetStatsCalculator()
[all …]
Dprofile_summary_formatter.cc49 auto subgraph_index = stats_calc.first; in GenerateReport() local
52 if (subgraph_index == 0) { in GenerateReport()
55 stream << "Subgraph (index: " << subgraph_index << ") " << tag << ":" in GenerateReport()
62 if (subgraph_index != 0) { in GenerateReport()
63 stream << "Subgraph (index: " << subgraph_index << ") "; in GenerateReport()
Dprofile_summarizer.h55 tensorflow::StatsCalculator* GetStatsCalculator(uint32_t subgraph_index);
/external/tensorflow/tensorflow/lite/tools/
Dlist_flex_ops_no_kernel.cc34 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() local
35 ++subgraph_index) { in AddFlexOpsFromModel()
36 const tflite::SubGraph* subgraph = subgraphs->Get(subgraph_index); in AddFlexOpsFromModel()
Dlist_flex_ops.cc82 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() local
83 ++subgraph_index) { in AddFlexOpsFromModel()
84 const tflite::SubGraph* subgraph = subgraphs->Get(subgraph_index); in AddFlexOpsFromModel()
/external/tensorflow/tensorflow/lite/
Dgraph_info_test.cc112 for (size_t subgraph_index = 0; subgraph_index < generated_subgraphs.size(); in CheckPartitionSubgraphs() local
113 subgraph_index++) { in CheckPartitionSubgraphs()
114 EXPECT_EQ(generated_subgraphs[subgraph_index].nodes, in CheckPartitionSubgraphs()
115 expected_subgraphs[subgraph_index].nodes); in CheckPartitionSubgraphs()
116 EXPECT_EQ(generated_subgraphs[subgraph_index].input_tensors, in CheckPartitionSubgraphs()
117 expected_subgraphs[subgraph_index].input_tensors); in CheckPartitionSubgraphs()
118 EXPECT_EQ(generated_subgraphs[subgraph_index].output_tensors, in CheckPartitionSubgraphs()
119 expected_subgraphs[subgraph_index].output_tensors); in CheckPartitionSubgraphs()
Dinterpreter.cc461 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in SetSubgraphProfiler() local
462 ++subgraph_index) { in SetSubgraphProfiler()
463 subgraphs_[subgraph_index]->SetProfiler(installed_profiler_, in SetSubgraphProfiler()
464 subgraph_index); in SetSubgraphProfiler()
Dinterpreter_builder.cc731 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in operator ()() local
732 ++subgraph_index) { in operator ()()
733 const tflite::SubGraph* subgraph = (*subgraphs)[subgraph_index]; in operator ()()
735 (*interpreter)->subgraph(subgraph_index); in operator ()()
741 subgraph_index); in operator ()()
Dinterpreter.h639 Subgraph* subgraph(int subgraph_index) { in subgraph() argument
640 if (subgraph_index < 0 || in subgraph()
641 static_cast<size_t>(subgraph_index) >= subgraphs_size()) in subgraph()
643 return &*subgraphs_[subgraph_index]; in subgraph()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc199 int subgraph_index, bool use_updated_hybrid_scheme) { in InsertQuantizableInputTensorsFromOperator() argument
200 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get(); in InsertQuantizableInputTensorsFromOperator()
430 for (int subgraph_index = 0, end = model->subgraphs.size(); in QuantizeWeightsInt8() local
431 subgraph_index < end; ++subgraph_index) { in QuantizeWeightsInt8()
432 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get(); in QuantizeWeightsInt8()
439 subgraph_index, use_updated_hybrid_scheme)); in QuantizeWeightsInt8()
554 for (int subgraph_index = 0, end = model->subgraphs.size(); in QuantizeWeightsFloat16() local
555 subgraph_index < end; ++subgraph_index) { in QuantizeWeightsFloat16()
556 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get(); in QuantizeWeightsFloat16()
Dmodify_model_interface.cc39 size_t subgraph_index; // index of the subgraph. member
200 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in SetInputTypeToUINT8()
227 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in SetOutputTypeToUINT8()
260 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in RemoveInputTensor()
286 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in RemoveOutputTensor()
Doperator_property.h143 OperatorProperty GetOperatorProperty(const ModelT* model, int subgraph_index,
Doperator_property.cc25 const OpVariant GetOperatorVariant(const ModelT* model, int subgraph_index, in GetOperatorVariant() argument
29 model->subgraphs.at(subgraph_index)->operators[op_index].get(); in GetOperatorVariant()
55 OperatorProperty GetOperatorProperty(const ModelT* model, int subgraph_index, in GetOperatorProperty() argument
57 OpVariant op_variant = GetOperatorVariant(model, subgraph_index, op_index); in GetOperatorProperty()
Dquantize_model.cc56 int subgraph_index, int op_idx, const string& operator_name, in GetOperatorProperty() argument
59 operator_property::GetOperatorProperty(model, subgraph_index, op_idx); in GetOperatorProperty()
60 const SubGraphT* subgraph = model->subgraphs[subgraph_index].get(); in GetOperatorProperty()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dlabel_image.cc215 uint32_t subgraph_index, uint32_t op_index, in PrintProfilingInfo() argument
224 << subgraph_index << ", Node " << std::setw(3) in PrintProfilingInfo()
378 auto subgraph_index = profile_events[i]->extra_event_metadata; in RunInference() local
380 const auto subgraph = interpreter->subgraph(subgraph_index); in RunInference()
384 PrintProfilingInfo(profile_events[i], subgraph_index, op_index, in RunInference()
/external/tensorflow/tensorflow/lite/tools/serialization/
DREADME.md31 // Note that 0 <= subgraph_index < num_subgraphs
32 tflite::SubgraphWriter writer(&interpreter->subgraph(subgraph_index));
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test.cc780 for (int subgraph_index = 0; subgraph_index < 2; subgraph_index++) { in TEST_F() local
781 ASSERT_EQ(interpreter_->subgraph(subgraph_index)->execution_plan().size(), in TEST_F()
783 int node = interpreter_->subgraph(subgraph_index)->execution_plan()[0]; in TEST_F()
785 interpreter_->subgraph(subgraph_index)->node_and_registration(node); in TEST_F()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h361 SubgraphAwareProfiler(Profiler* profiler, int64_t subgraph_index) in SubgraphAwareProfiler() argument
362 : profiler_(profiler), subgraph_index_(subgraph_index) {} in SubgraphAwareProfiler()