| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 73 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildAddSubgraph()
103 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildMulSubgraph()
133 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildPadSubgraph()
163 ASSERT_EQ(subgraph->SetInputs({kCondInput, kInput1, kInput2}), kTfLiteOk); in BuildIfSubgraph()
203 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildLessEqualCondSubgraph()
238 ASSERT_EQ(subgraph->SetInputs({kInputCounter, kInputValue}), kTfLiteOk); in BuildAccumulateLoopBodySubgraph()
283 ASSERT_EQ(subgraph->SetInputs({kInputCounter, kInputValue}), kTfLiteOk); in BuildPadLoopBodySubgraph()
326 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildWhileSubgraph()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 65 interpreter.SetInputs({}); in TEST()
123 ASSERT_NE(interpreter.SetInputs({5}), kTfLiteOk); in TEST()
151 ASSERT_EQ(interpreter.SetInputs({0}), kTfLiteOk); in TEST()
173 interpreter.SetInputs({0, 1}); in TEST()
190 interpreter.SetInputs({0, 1}); in TEST()
257 interpreter.SetInputs({0, 1}); in TEST()
321 interpreter.SetInputs({0, 1}); in TEST()
346 ASSERT_EQ(interpreter.SetInputs({0}), kTfLiteOk); in TEST()
363 ASSERT_EQ(interpreter.SetInputs({0}), kTfLiteOk); in TEST()
379 ASSERT_EQ(interpreter.SetInputs({0}), kTfLiteOk); in TEST()
[all …]
|
| D | interpreter.cc | 86 TfLiteStatus Interpreter::SetInputs(std::vector<int> inputs) { in SetInputs() function in tflite::Interpreter
87 return primary_subgraph().SetInputs(inputs); in SetInputs()
|
| D | interpreter.h | 125 TfLiteStatus SetInputs(std::vector<int> inputs);
|
| D | model.cc | 542 modified_subgraph->SetInputs( in operator ()()
|
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer_test.cc | 78 void SetInputs(int32_t x, int32_t y) { in SetInputs() function in tflite::profiling::__anonfc60a69d0111::SimpleOpModel
112 m.SetInputs(1, 2); in TEST()
133 m.SetInputs(1, 2); in TEST()
|
| /external/tensorflow/tensorflow/lite/experimental/writer/ |
| D | writer_lib_test.cc | 37 interpreter.SetInputs({0, 1}); in TEST()
|
| /external/tensorflow/tensorflow/lite/examples/label_image/ |
| D | bitmap_helpers_impl.h | 44 interpreter->SetInputs({0, 1}); in resize()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.h | 140 void SetInputs(std::vector<std::pair<TF_Operation*, TF_Tensor*>> inputs);
|
| D | while_loop_test.cc | 90 csession_->SetInputs(inputs); in Run()
|
| D | c_api_function_test.cc | 96 csession.SetInputs(inputs); in RunT()
120 csession.SetInputs(inputs); in Run()
1711 csession.SetInputs({{feed, Int32Tensor(17)}}); in TEST_F()
|
| D | c_api_test.cc | 961 csession.SetInputs({{feed, Int32Tensor(3)}}); in TEST()
978 csession.SetInputs({{feed, Int32Tensor(7)}}); in TEST()
1022 csession.SetInputs({{feed, Int32Tensor({3, 2, 5})}}); in RunMinTest()
1396 csession.SetInputs({{input_op, TF_TensorFromTensor(input, s)}}); in TEST()
|
| D | c_test_util.cc | 469 void CSession::SetInputs( in SetInputs() function in CSession
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | test_util.cc | 87 CHECK_EQ(interpreter_->SetInputs(inputs), kTfLiteOk); in AddTensors()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | parallel_interleave_dataset_op.cc | 325 current_worker->SetInputs(s, std::move(args)); in GetNextInternal()
543 void SetInputs(const Status& s, std::vector<Tensor> input_arguments) { in SetInputs() function
593 workers_[i].SetInputs(s, std::move(args)); in EnsureWorkerThreadsStarted()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 50 TfLiteStatus SetInputs(std::vector<int> inputs);
|
| D | subgraph.cc | 373 TfLiteStatus Subgraph::SetInputs(std::vector<int> inputs) { in SetInputs() function in tflite::Subgraph
|