Searched refs:execution_plan (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/java/src/test/native/ |
| D | interpreter_test_jni.cc | 56 TfLiteIntArray* execution_plan; in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate() local
58 context->GetExecutionPlan(context, &execution_plan)); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate()
60 context, registration, execution_plan, delegate); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 746 TfLiteIntArray* execution_plan; in TEST() local
747 EXPECT_EQ(context->GetExecutionPlan(context, &execution_plan), in TEST()
1134 TfLiteIntArray* execution_plan; in SimpleDelegate() local
1136 context->GetExecutionPlan(context, &execution_plan)); in SimpleDelegate()
1137 for (int exec_index = 0; exec_index < execution_plan->size; in SimpleDelegate()
1139 int node_index = execution_plan->data[exec_index]; in SimpleDelegate()
1196 ASSERT_EQ(interpreter_->execution_plan().size(), 1); in TEST_F()
1197 int node = interpreter_->execution_plan()[0]; in TEST_F()
1224 ASSERT_EQ(interpreter_->execution_plan().size(), 1); in TEST_F()
1238 ASSERT_EQ(interpreter_->execution_plan().size(), 2); in TEST_F()
[all …]
|
| D | interpreter.h | 241 const std::vector<int>& execution_plan() const { in execution_plan() function
242 return primary_subgraph().execution_plan(); in execution_plan()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 167 std::vector<int>& execution_plan() { return execution_plan_; } in execution_plan() function
170 const std::vector<int>& execution_plan() const { return execution_plan_; } in execution_plan() function
409 TfLiteStatus GetExecutionPlan(TfLiteIntArray** execution_plan);
414 TfLiteIntArray** execution_plan);
|
| D | subgraph.cc | 119 return subgraph_->execution_plan().size(); in num_nodes()
122 int node_index = subgraph_->execution_plan()[index]; in node()
354 TfLiteStatus Subgraph::GetExecutionPlan(TfLiteIntArray** execution_plan) { in GetExecutionPlan() argument
357 *execution_plan = plan_cache_.get(); in GetExecutionPlan()
368 TfLiteIntArray** execution_plan) { in GetExecutionPlan() argument
370 ->GetExecutionPlan(execution_plan); in GetExecutionPlan()
|
| /external/tensorflow/tensorflow/lite/experimental/writer/ |
| D | writer_lib.cc | 55 for (int op_index : interpreter_->execution_plan()) { in ExportOperators()
68 for (int op_index : interpreter_->execution_plan()) { in ExportOperators()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_internal.h | 413 TfLiteIntArray** execution_plan);
|