| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test.cc | 55 ASSERT_EQ(interpreter_->execution_plan().size(), 1); in TEST_F()
56 int node = interpreter_->execution_plan()[0]; in TEST_F()
86 ASSERT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
110 ASSERT_EQ(interpreter_->execution_plan().size(), 1); in TEST_F()
122 ASSERT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
145 ASSERT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
151 ASSERT_EQ(interpreter_->execution_plan().size(), 2); in TEST_F()
158 ASSERT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
186 ASSERT_EQ(interpreter_->execution_plan().size(), 2); in TEST_F()
198 ASSERT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
[all …]
|
| D | interpreter_utils_test.cc | 45 ASSERT_EQ(interpreter_->execution_plan().size(), 1); in TEST_F()
57 ASSERT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
76 ASSERT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
84 ASSERT_EQ(interpreter_->execution_plan().size(), 2); in TEST_F()
97 EXPECT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
127 ASSERT_EQ(interpreter_->execution_plan().size(), 8); in TEST_P()
|
| D | utils.cc | 116 TfLiteIntArray* execution_plan = nullptr; in PrepareSupportedNodes() local
117 auto status = context_->GetExecutionPlan(context_, &execution_plan); in PrepareSupportedNodes()
123 num_total_nodes_ = execution_plan->size; in PrepareSupportedNodes()
126 for (int node_id : TfLiteIntArrayView(execution_plan)) { in PrepareSupportedNodes()
|
| D | utils_test.cc | 133 TfLiteIntArray** execution_plan) { in MockGetExecutionPlan() argument
135 *execution_plan = mock->exec_plan(); in MockGetExecutionPlan()
212 TfLiteIntArray** execution_plan) { in ErrorGetExecutionPlan() argument
|
| D | delegate_test_util.cc | 161 TfLiteIntArray* execution_plan; in SimpleDelegate() local
162 TF_LITE_ENSURE_STATUS(context->GetExecutionPlan(context, &execution_plan)); in SimpleDelegate()
163 for (int exec_index = 0; exec_index < execution_plan->size; exec_index++) { in SimpleDelegate()
164 int node_index = execution_plan->data[exec_index]; in SimpleDelegate()
|
| /external/tensorflow/tensorflow/lite/ |
| D | model_xnnpack_test.cc | 48 EXPECT_EQ(1, interpreter->execution_plan().size()); in TEST()
49 int first_node_id = interpreter->execution_plan()[0]; in TEST()
76 EXPECT_LT(1, interpreter->execution_plan().size()); in TEST()
77 int first_node_id = interpreter->execution_plan()[0]; in TEST()
|
| D | interpreter.h | 237 const std::vector<int>& execution_plan() const { in execution_plan() function
238 return primary_subgraph().execution_plan(); in execution_plan()
|
| /external/tensorflow/tensorflow/lite/java/src/test/native/ |
| D | interpreter_test_jni.cc | 62 TfLiteIntArray* execution_plan; in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate() local
64 context->GetExecutionPlan(context, &execution_plan)); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate()
66 context, registration, execution_plan, delegate); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | test_util.mm | 44 XCTAssertEqual(interpreter->execution_plan().size(), 1);
45 const int node_index = interpreter->execution_plan()[0];
57 for (int node_idx : interpreter->execution_plan()) {
59 XCTAssertEqual(interpreter->execution_plan().size(), 1);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 303 TfLiteIntArray** execution_plan) { in TEST() argument
304 *execution_plan = interpreter_fp16_add_op->exec_plan(); in TEST()
362 TfLiteIntArray** execution_plan) { in TEST() argument
363 *execution_plan = interpreter_fp16_non_constant->exec_plan(); in TEST()
425 TfLiteIntArray** execution_plan) { in TEST() argument
426 *execution_plan = interpreter_fp16_gt_op->exec_plan(); in TEST()
539 TfLiteIntArray** execution_plan) { in TEST() argument
540 *execution_plan = interpreter_fp32->exec_plan(); in TEST()
706 TfLiteIntArray** execution_plan) { in TEST() argument
707 *execution_plan = interpreter2_fp32->exec_plan(); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib.h | 92 execution_plan_(subgraph->execution_plan()) { in SubgraphWriter()
115 const std::vector<int>& execution_plan);
127 execution_plan_(subgraph->execution_plan()) { in SubgraphWriter()
161 const std::vector<int>& execution_plan);
|
| D | writer_lib.cc | 329 const std::vector<int>& execution_plan) { in CheckInputOutput() argument
335 for (int op_index : execution_plan) { in CheckInputOutput()
388 const std::vector<int>& execution_plan) { in SetCustomInputOutput() argument
389 TF_LITE_ENSURE_STATUS(CheckInputOutput(inputs, outputs, execution_plan)); in SetCustomInputOutput()
392 execution_plan_ = execution_plan; in SetCustomInputOutput()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | hexagon_delegate_op_model.h | 66 ASSERT_EQ(1, interpreter_->execution_plan().size()); in ApplyDelegateAndInvoke()
67 const int node = interpreter_->execution_plan()[0]; in ApplyDelegateAndInvoke()
|
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate_test.cc | 69 ASSERT_EQ(interpreter_->execution_plan().size(), 1); in TEST_F()
70 int node = interpreter_->execution_plan()[0]; in TEST_F()
97 ASSERT_EQ(interpreter_->execution_plan().size(), 3); in TEST_F()
|
| /external/XNNPACK/bench/ |
| D | end2end.cc | 33 auto execution_plan = model_factory(threadpool.get()); in End2EndBenchmark() local
34 if (execution_plan.empty()) { in End2EndBenchmark()
40 for (const std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)>& op : execution_plan) { in End2EndBenchmark()
|
| D | f32-dwconv-e2e.cc | 51 auto execution_plan = model_factory(nullptr); in DWConvEnd2EndBenchmark() local
52 if (execution_plan.empty()) { in DWConvEnd2EndBenchmark()
58 for (const std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)>& op : execution_plan) { in DWConvEnd2EndBenchmark()
|
| D | f32-gemm-e2e.cc | 53 auto execution_plan = model_factory(nullptr); in GEMMEnd2EndBenchmark() local
54 if (execution_plan.empty()) { in GEMMEnd2EndBenchmark()
60 for (const std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)>& op : execution_plan) { in GEMMEnd2EndBenchmark()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 192 std::vector<int>& execution_plan() { return execution_plan_; } in execution_plan() function
195 const std::vector<int>& execution_plan() const { return execution_plan_; } in execution_plan() function
441 const std::vector<int>& execution_plan,
519 TfLiteStatus GetExecutionPlan(TfLiteIntArray** execution_plan);
524 TfLiteIntArray** execution_plan);
|
| D | subgraph.cc | 185 return subgraph_->execution_plan().size(); in num_execution_nodes()
189 int node_index = subgraph_->execution_plan()[index]; in node()
193 return subgraph_->execution_plan()[index]; in node_index()
469 TfLiteStatus Subgraph::GetExecutionPlan(TfLiteIntArray** execution_plan) { in GetExecutionPlan() argument
472 *execution_plan = plan_cache_.get(); in GetExecutionPlan()
483 TfLiteIntArray** execution_plan) { in GetExecutionPlan() argument
485 ->GetExecutionPlan(execution_plan); in GetExecutionPlan()
901 int first_execution_plan_index, const std::vector<int>& execution_plan, in PrepareOpsStartingAt() argument
910 execution_plan_index < execution_plan.size(); execution_plan_index++) { in PrepareOpsStartingAt()
911 int node_index = execution_plan[execution_plan_index]; in PrepareOpsStartingAt()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 412 const Tensor& execution_plan = context->input(0); in DoWork() local
413 TF_RET_CHECK(TensorShapeUtils::IsScalar(execution_plan.shape())); in DoWork()
415 TF_RET_CHECK(plan.ParseFromString(execution_plan.scalar<tstring>()())); in DoWork()
|
| D | xrt_execute_op.cc | 530 const Tensor& execution_plan = context->input(0); in DoWork() local
531 TF_RET_CHECK(TensorShapeUtils::IsScalar(execution_plan.shape())); in DoWork()
533 TF_RET_CHECK(ParseFromTString(execution_plan.scalar<tstring>()(), &plan)); in DoWork()
|
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| D | benchmark_tflite_model.cc | 643 for (int i = 0; i < interpreter_->execution_plan().size(); ++i) { in Init()
644 int node_id = interpreter_->execution_plan()[i]; in Init()
661 interpreter_->execution_plan().size() == 1); in Init()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 121 TfLiteIntArray* execution_plan; in Create() local
122 if (context->GetExecutionPlan(context, &execution_plan) != kTfLiteOk) { in Create()
2931 TfLiteIntArray* execution_plan = nullptr; in PrepareOpsToDelegate() local
2932 if (context->GetExecutionPlan(context, &execution_plan) != kTfLiteOk) { in PrepareOpsToDelegate()
2946 TfLiteIntArrayCreate(execution_plan->size); in PrepareOpsToDelegate()
2948 for (int i = 0; i < execution_plan->size; ++i) { in PrepareOpsToDelegate()
2949 const int node_index = execution_plan->data[i]; in PrepareOpsToDelegate()
3241 nodes_to_delegate->size = execution_plan->size; in PrepareOpsToDelegate()
3242 std::copy(&execution_plan->data[0], in PrepareOpsToDelegate()
3243 &execution_plan->data[execution_plan->size], in PrepareOpsToDelegate()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 641 TfLiteIntArray** execution_plan);
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | nativeinterpreterwrapper_jni.cc | 259 for (int node_i : subgraph->execution_plan()) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_hasUnresolvedFlexOp()
360 return static_cast<jint>(interpreter->execution_plan().size()); in Java_org_tensorflow_lite_NativeInterpreterWrapper_getExecutionPlanLength()
|