Home
last modified time | relevance | path

Searched refs:num_execution_nodes (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dgraph_info.cc52 num_incoming_control_edges_(info_->num_execution_nodes(), 0) { in PartitionGraphIntoIndependentNodeSubsetsImpl()
67 node_epochs_.resize(info_->num_execution_nodes(), kEpochNotReady); in Partition()
69 num_incoming_control_edges_.resize(info_->num_execution_nodes(), 0); in Partition()
76 for (int node_index = 0; node_index < info_->num_execution_nodes(); in Partition()
212 for (int node_index = 0; node_index < info_->num_execution_nodes(); in BuildNodeSubset()
264 node_index < info->num_execution_nodes(); ++node_index) { in PartitionGraphIntoIndependentNodeSubsets()
H A Dsimple_planner.cc129 for (size_t i = 0; i < graph_info_->num_execution_nodes(); ++i) { in PlanAllocations()
141 for (size_t i = 0; i < graph_info_->num_execution_nodes(); ++i) { in PlanAllocations()
176 i < graph_info_->num_execution_nodes(); in ExecuteAllocations()
H A Darena_planner.cc147 for (size_t i = 0; i < graph_info_->num_execution_nodes(); ++i) { in PlanAllocations()
159 for (size_t i = 0; i < graph_info_->num_execution_nodes(); ++i) { in PlanAllocations()
199 i < graph_info_->num_execution_nodes(); in ExecuteAllocations()
H A Dgraph_info.h42 virtual size_t num_execution_nodes() const = 0;
H A Dsimple_planner_test.cc140 size_t num_execution_nodes() const override { return graph_->nodes().size(); } in num_execution_nodes() function in tflite::__anonec026e940111::TestGraphInfo
H A Dgraph_info_test.cc55 size_t num_execution_nodes() const override { in num_execution_nodes() function in tflite::__anonb7acd6780111::SimpleTestGraph
H A Darena_planner_test.cc160 size_t num_execution_nodes() const override { return graph_->nodes().size(); } in num_execution_nodes() function in tflite::__anone7daca910111::TestGraphInfo
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc206 size_t num_execution_nodes() const override { in num_execution_nodes() function in tflite::InterpreterInfo
1201 for (size_t i = 0; i < graph_info->num_execution_nodes(); ++i) { in RemoveUnusedInputs()