Home
last modified time | relevance | path

Searched refs:AddNodeWithParameters (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dvariable_ops_test.cc67 interpreter_->AddNodeWithParameters({}, {1}, nullptr, 0, var_handle_params, in ConstructInvalidGraph()
69 interpreter_->AddNodeWithParameters({1}, {2}, nullptr, 0, nullptr, in ConstructInvalidGraph()
104 interpreter_->AddNodeWithParameters({}, {1}, nullptr, 0, var_handle_params, in ConstructGraph()
106 interpreter_->AddNodeWithParameters({1, 0}, {}, nullptr, 0, nullptr, in ConstructGraph()
108 interpreter_->AddNodeWithParameters({1}, {2}, nullptr, 0, nullptr, in ConstructGraph()
135 interpreter_->AddNodeWithParameters({}, {1}, nullptr, 0, var_handle_params, in ConstructGraphWithKnownShape()
137 interpreter_->AddNodeWithParameters({1, 0}, {}, nullptr, 0, nullptr, in ConstructGraphWithKnownShape()
139 interpreter_->AddNodeWithParameters({1}, {2}, nullptr, 0, nullptr, in ConstructGraphWithKnownShape()
Dsubgraph_test_util.cc116 subgraph->AddNodeWithParameters({kInput1, kInput2}, {kOutput}, {}, nullptr, 0, in BuildAddSubgraph()
147 subgraph->AddNodeWithParameters({kInput1, kInput2}, {kOutput}, {}, nullptr, 0, in BuildMulSubgraph()
177 subgraph->AddNodeWithParameters({kInput1, kInput2}, {kOutput}, {}, nullptr, 0, in BuildPadSubgraph()
212 subgraph->AddNodeWithParameters({kCondInput, kInput1, kInput2}, {kOutput}, {}, in BuildIfSubgraph()
245 subgraph->AddNodeWithParameters({kInput1, kConstRhs}, {kOutput}, {}, nullptr, in BuildLessEqualCondSubgraph()
286 subgraph->AddNodeWithParameters({0, 4}, {2}, {}, nullptr, 0, params, add_reg, in BuildAccumulateLoopBodySubgraph()
291 subgraph->AddNodeWithParameters({2, 1}, {3}, {}, nullptr, 0, params, add_reg, in BuildAccumulateLoopBodySubgraph()
337 subgraph->AddNodeWithParameters({kInputCounter, kConstStep}, {kOutputCounter}, in BuildPadLoopBodySubgraph()
344 subgraph->AddNodeWithParameters({kInputValue, kConstPadding}, {kOutputValue}, in BuildPadLoopBodySubgraph()
380 subgraph->AddNodeWithParameters({0, 1}, {2, 3}, {}, nullptr, 0, params, in BuildWhileSubgraph()
[all …]
Dhashtable_ops_test.cc122 interpreter_->AddNodeWithParameters( in BuildDefaultGraph()
128 interpreter_->AddNodeWithParameters( in BuildDefaultGraph()
133 interpreter_->AddNodeWithParameters( in BuildDefaultGraph()
139 interpreter_->AddNodeWithParameters( in BuildDefaultGraph()
149 interpreter_->AddNodeWithParameters( in BuildNoImportGraph()
155 interpreter_->AddNodeWithParameters( in BuildNoImportGraph()
161 interpreter_->AddNodeWithParameters( in BuildNoImportGraph()
171 interpreter_->AddNodeWithParameters( in BuildImportTwiceGraph()
177 interpreter_->AddNodeWithParameters( in BuildImportTwiceGraph()
182 interpreter_->AddNodeWithParameters( in BuildImportTwiceGraph()
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph_test.cc43 subgraph.AddNodeWithParameters({0, 1}, {2}, {}, nullptr, 0, nullptr, pad_op); in TEST()
44 subgraph.AddNodeWithParameters({2}, {3}, {}, nullptr, 0, nullptr, neg_op); in TEST()
57 subgraph.AddNodeWithParameters({2}, {3}, {}, nullptr, 0, nullptr, neg_op); in TEST()
Dsubgraph.h91 TfLiteStatus AddNodeWithParameters(const std::vector<int>& inputs,
Dsubgraph.cc448 TF_LITE_ENSURE_STATUS(AddNodeWithParameters( in ReplaceNodeSubsetsWithDelegateKernels()
792 TfLiteStatus Subgraph::AddNodeWithParameters( in AddNodeWithParameters() function in tflite::Subgraph
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc110 subgraph->AddNodeWithParameters({0, 0}, {2}, {}, nullptr, 0, nullptr, &reg, in SetUpSubgraph()
112 subgraph->AddNodeWithParameters({1, 1}, {3}, {}, nullptr, 0, nullptr, &reg, in SetUpSubgraph()
114 subgraph->AddNodeWithParameters({2, 1}, {4}, {}, nullptr, 0, nullptr, &reg, in SetUpSubgraph()
440 interpreter_->AddNodeWithParameters({1}, {2}, nullptr, 0, nullptr, deq_reg); in SetUp()
441 interpreter_->AddNodeWithParameters({0, 2}, {3}, nullptr, 0, builtin_data0, in SetUp()
443 interpreter_->AddNodeWithParameters({4}, {5}, nullptr, 0, nullptr, deq_reg); in SetUp()
444 interpreter_->AddNodeWithParameters({3, 5}, {6}, nullptr, 0, builtin_data1, in SetUp()
446 interpreter_->AddNodeWithParameters({7}, {8}, nullptr, 0, nullptr, deq_reg); in SetUp()
447 interpreter_->AddNodeWithParameters({6, 8}, {9}, nullptr, 0, builtin_data2, in SetUp()
449 interpreter_->AddNodeWithParameters({10}, {11}, nullptr, 0, nullptr, deq_reg); in SetUp()
[all …]
Ddelegate_test.cc869 interpreter_->AddNodeWithParameters({0}, {1, 2}, nullptr, 0, nullptr, &reg); in SetUp()
1014 interpreter_->AddNodeWithParameters({0}, {1}, nullptr, 0, nullptr, &reg); in SetUp()
1015 interpreter_->AddNodeWithParameters({1}, {2}, nullptr, 0, nullptr, &reg); in SetUp()
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc85 interpreter.AddNodeWithParameters({0, 1}, {2}, initial_data, 0, in TEST_P()
128 interpreter.AddNodeWithParameters({0, 1}, {2}, initial_data, 0, in TEST_P()
166 interpreter.AddNodeWithParameters({0, 1}, {2}, initial_data, 0, in TEST_P()
170 interpreter.AddNodeWithParameters({2}, {3}, nullptr, 0, nullptr, reg2); in TEST_P()
215 interpreter.AddNodeWithParameters({0, 1}, {2}, initial_data, 0, in TEST_P()
219 interpreter.AddNodeWithParameters({2}, {3}, nullptr, 0, nullptr, reg2); in TEST_P()
222 interpreter.AddNodeWithParameters({3}, {4}, nullptr, 0, nullptr, reg3); in TEST_P()
263 interpreter.AddNodeWithParameters({0, 1}, {2}, nullptr, 0, in TEST_P()
302 interpreter.AddNodeWithParameters({0, 1}, {2}, initial_data, 0, in TEST_P()
330 interpreter.AddNodeWithParameters(/*inputs=*/{0, 1}, /*outputs=*/{2}, in TEST_P()
[all …]
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc170 ASSERT_NE(interpreter.AddNodeWithParameters({3}, {0}, nullptr, 0, nullptr, in TEST()
183 ASSERT_EQ(interpreter.AddNodeWithParameters({0}, {1}, nullptr, 0, nullptr, in TEST()
362 interpreter.AddNodeWithParameters({0, 1}, {2, 3}, nullptr, 0, nullptr, &reg); in TEST()
363 interpreter.AddNodeWithParameters({2, 1}, {4, 5}, nullptr, 0, nullptr, &reg); in TEST()
364 interpreter.AddNodeWithParameters({4, 3}, {6, 7}, nullptr, 0, nullptr, &reg); in TEST()
365 interpreter.AddNodeWithParameters({6, 5}, {8}, nullptr, 0, nullptr, &reg); in TEST()
366 interpreter.AddNodeWithParameters({8, 7}, {9}, nullptr, 0, nullptr, &reg); in TEST()
438 interpreter.AddNodeWithParameters({0}, {1}, nullptr, 0, nullptr, &reg); in TEST()
693 interpreter.AddNodeWithParameters({0}, {1}, nullptr, 0, nullptr, &reg), in TEST()
718 interpreter.AddNodeWithParameters({0}, {1}, nullptr, 0, nullptr, &reg), in TEST()
[all …]
Dinterpreter.cc202 TfLiteStatus Interpreter::AddNodeWithParameters( in AddNodeWithParameters() function in tflite::Interpreter
206 return primary_subgraph().AddNodeWithParameters( in AddNodeWithParameters()
Dinterpreter_builder.cc359 subgraph->AddNodeWithParameters( in ParseNodes()
366 subgraph->AddNodeWithParameters( in ParseNodes()
377 subgraph->AddNodeWithParameters( in ParseNodes()
Dinterpreter.h140 TfLiteStatus AddNodeWithParameters(const std::vector<int>& inputs,
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc50 interpreter_->AddNodeWithParameters({0, 0}, {2}, nullptr, 0, builtin_data_1, in SetUp()
52 interpreter_->AddNodeWithParameters({1, 1}, {3}, nullptr, 0, builtin_data_2, in SetUp()
54 interpreter_->AddNodeWithParameters({2, 1}, {4}, nullptr, 0, builtin_data_3, in SetUp()
/external/armnn/delegate/classic/src/test/
DArmnnClassicDelegateTest.cpp36 tfLiteInterpreter->AddNodeWithParameters({0, 1}, {2}, "", 0, nullptr, opRegister);
74 tfLiteInterpreter->AddNodeWithParameters({0, 1}, {2}, "", 0, nullptr, opRegister);
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Dcall_test.cc65 subgraph->AddNodeWithParameters( in BuildCallSubgraph()
115 subgraph->AddNodeWithParameters( in BuildGraphWithMultipleOutputs()
121 subgraph->AddNodeWithParameters( in BuildGraphWithMultipleOutputs()
155 this_subgraph->AddNodeWithParameters( in BuildMultiNodeGraph()
174 this_subgraph->AddNodeWithParameters({kInput4, kOutput2}, {kOutput3}, {}, in BuildMultiNodeGraph()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc216 EXPECT_EQ(interpreter_.AddNodeWithParameters( in InterpreterFp16()
225 EXPECT_EQ(interpreter_.AddNodeWithParameters( in InterpreterFp16()
243 EXPECT_EQ(interpreter_.AddNodeWithParameters( in InterpreterFp16()
491 EXPECT_EQ(interpreter_.AddNodeWithParameters( in InterpreterFp32()
509 EXPECT_EQ(interpreter_.AddNodeWithParameters( in InterpreterFp32()
606 EXPECT_EQ(interpreter_.AddNodeWithParameters( in Interpreter2Fp32()
624 EXPECT_EQ(interpreter_.AddNodeWithParameters( in Interpreter2Fp32()
638 EXPECT_EQ(interpreter_.AddNodeWithParameters( in Interpreter2Fp32()
655 EXPECT_EQ(interpreter_.AddNodeWithParameters( in Interpreter2Fp32()
785 EXPECT_EQ(interpreter_.AddNodeWithParameters( in InterpreterMultiNode()
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc123 CHECK_EQ(interpreter_->AddNodeWithParameters(inputs, outputs, nullptr, 0, in AddTfLiteMulOp()
208 CHECK_EQ(interpreter_->AddNodeWithParameters( in AddTfOp()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h64 interpreter->AddNodeWithParameters({0, 1}, {2}, nullptr, 0, params, resize_op, in resize()