| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 74 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildAddSubgraph()
104 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildMulSubgraph()
134 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildPadSubgraph()
164 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildIfSubgraph()
204 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildLessEqualCondSubgraph()
239 ASSERT_EQ(subgraph->SetOutputs({kOutputCounter, kOutputValue}), kTfLiteOk); in BuildAccumulateLoopBodySubgraph()
284 ASSERT_EQ(subgraph->SetOutputs({kOutputCounter, kOutputValue}), kTfLiteOk); in BuildPadLoopBodySubgraph()
327 ASSERT_EQ(subgraph->SetOutputs({kOutput1, kOutput2}), kTfLiteOk); in BuildWhileSubgraph()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 66 interpreter.SetOutputs({}); in TEST()
131 ASSERT_NE(interpreter.SetOutputs({5}), kTfLiteOk); in TEST()
152 ASSERT_EQ(interpreter.SetOutputs({0}), kTfLiteOk); in TEST()
174 interpreter.SetOutputs({}); in TEST()
191 interpreter.SetOutputs({}); in TEST()
258 interpreter.SetOutputs({}); in TEST()
322 interpreter.SetOutputs({9, 4}); in TEST()
364 ASSERT_EQ(interpreter.SetOutputs({0}), kTfLiteOk); in TEST()
399 interpreter.SetOutputs({1}); in TEST()
431 ASSERT_EQ(interpreter.SetOutputs({0}), kTfLiteOk); in TEST()
[all …]
|
| D | interpreter.cc | 90 TfLiteStatus Interpreter::SetOutputs(std::vector<int> outputs) { in SetOutputs() function in tflite::Interpreter
91 return primary_subgraph().SetOutputs(outputs); in SetOutputs()
|
| D | interpreter.h | 130 TfLiteStatus SetOutputs(std::vector<int> outputs);
|
| D | model.cc | 544 modified_subgraph->SetOutputs( in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 98 Status SetOutputs(const OpKernel* kernel, OpKernelContext* ctx, in SetOutputs() function
192 s = SetOutputs(kernel_, ctx_, rets_); in Start()
285 s = SetOutputs(kernel_, ctx_, rets_); in Start()
469 s = SetOutputs(kernel_, ctx_, args_); in Finish()
610 s = SetOutputs(kernel_, ctx_, rets_); in Finish()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.h | 141 void SetOutputs(std::initializer_list<TF_Operation*> outputs);
142 void SetOutputs(const std::vector<TF_Output>& outputs);
|
| D | c_api_experimental_test.cc | 43 csession.SetOutputs({get_next}); in TestFakeIteratorStack()
89 csession.SetOutputs({{get_next, 0}, {get_next, 1}}); in TEST()
|
| D | c_test_util.cc | 479 void CSession::SetOutputs(std::initializer_list<TF_Operation*> outputs) { in SetOutputs() function in CSession
488 void CSession::SetOutputs(const std::vector<TF_Output>& outputs) { in SetOutputs() function in CSession
|
| D | c_api_test.cc | 962 csession.SetOutputs({add}); in TEST()
979 csession.SetOutputs({neg}); in TEST()
1023 csession.SetOutputs({min}); in RunMinTest()
1404 csession.SetOutputs({output_op}); in TEST()
1582 csession->SetOutputs(grad_outputs_vec); in RunGraphsAndCompareOutputs()
1591 expected_csession->SetOutputs(expected_grad_outputs_vec); in RunGraphsAndCompareOutputs()
|
| D | while_loop_test.cc | 91 csession_->SetOutputs(run_outputs); in Run()
|
| D | c_api_function_test.cc | 97 csession.SetOutputs(outputs); in RunT()
121 csession.SetOutputs(outputs); in Run()
1712 csession.SetOutputs({op}); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/experimental/writer/ |
| D | writer_lib_test.cc | 38 interpreter.SetOutputs({2}); in TEST()
|
| /external/tensorflow/tensorflow/lite/examples/label_image/ |
| D | bitmap_helpers_impl.h | 45 interpreter->SetOutputs({2}); in resize()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | test_util.cc | 88 CHECK_EQ(interpreter_->SetOutputs(outputs), kTfLiteOk); in AddTensors()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 55 TfLiteStatus SetOutputs(std::vector<int> outputs);
|
| D | subgraph.cc | 380 TfLiteStatus Subgraph::SetOutputs(std::vector<int> outputs) { in SetOutputs() function in tflite::Subgraph
|