| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 109 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildAddSubgraph()
140 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildMulSubgraph()
171 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildPadSubgraph()
202 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildIfSubgraph()
239 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildLessEqualCondSubgraph()
276 ASSERT_EQ(subgraph->SetOutputs({kOutputCounter, kOutputValue}), kTfLiteOk); in BuildAccumulateLoopBodySubgraph()
323 ASSERT_EQ(subgraph->SetOutputs({kOutputCounter, kOutputValue}), kTfLiteOk); in BuildPadLoopBodySubgraph()
370 ASSERT_EQ(subgraph->SetOutputs({kOutput1, kOutput2}), kTfLiteOk); in BuildWhileSubgraph()
403 ASSERT_EQ(subgraph->SetOutputs({}), kTfLiteOk); in BuildAssignRandomValueToVariableSubgraph()
430 ASSERT_EQ(subgraph->SetOutputs({kOutput}), kTfLiteOk); in BuildCallOnceAndReadVariableSubgraph()
[all …]
|
| D | variable_ops_test.cc | 56 ASSERT_EQ(interpreter_.SetOutputs({3}), kTfLiteOk); in ConstructGraph()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 72 interpreter.SetOutputs({}); in TEST()
158 ASSERT_NE(interpreter.SetOutputs({5}), kTfLiteOk); in TEST()
179 ASSERT_EQ(interpreter.SetOutputs({0}), kTfLiteOk); in TEST()
202 interpreter.SetOutputs({}); in TEST()
219 interpreter.SetOutputs({}); in TEST()
292 interpreter.SetOutputs({}); in TEST()
355 interpreter.SetOutputs({9, 4}); in TEST()
396 ASSERT_EQ(interpreter.SetOutputs({0}), kTfLiteOk); in TEST()
431 interpreter.SetOutputs({1}); in TEST()
463 ASSERT_EQ(interpreter.SetOutputs({0}), kTfLiteOk); in TEST()
[all …]
|
| D | interpreter.cc | 173 TfLiteStatus Interpreter::SetOutputs(std::vector<int> outputs) { in SetOutputs() function in tflite::Interpreter
174 return primary_subgraph().SetOutputs(std::move(outputs)); in SetOutputs()
|
| D | interpreter.h | 117 TfLiteStatus SetOutputs(std::vector<int> outputs);
|
| D | interpreter_builder.cc | 751 modified_subgraph->SetOutputs( in operator ()()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib_test.cc | 71 interpreter.SetOutputs({2}); in TEST_P()
114 interpreter.SetOutputs({2}); in TEST_P()
150 interpreter.SetOutputs({3}); in TEST_P()
199 interpreter.SetOutputs({4}); in TEST_P()
249 interpreter.SetOutputs({2}); in TEST_P()
288 interpreter.SetOutputs({2}); in TEST_P()
352 interpreter.SetOutputs({total_tensors - 1}); in TEST_P()
|
| /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_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 | while_loop_test.cc | 90 csession_->SetOutputs(run_outputs); in Run()
|
| D | c_api_test.cc | 999 csession.SetOutputs({add}); in TEST()
1016 csession.SetOutputs({neg}); in TEST()
1060 csession.SetOutputs({min}); in RunMinTest()
1442 csession.SetOutputs({output_op}); in TEST()
1620 csession->SetOutputs(grad_outputs_vec); in RunGraphsAndCompareOutputs()
1629 expected_csession->SetOutputs(expected_grad_outputs_vec); in RunGraphsAndCompareOutputs()
|
| D | c_api_function_test.cc | 97 csession.SetOutputs(outputs); in RunT()
121 csession.SetOutputs(outputs); in Run()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 96 Status SetOutputs(const OpKernel* kernel, OpKernelContext* ctx, in SetOutputs() function
184 s = SetOutputs(kernel_, ctx_, rets_); in Start()
314 s = SetOutputs(kernel_, ctx_, rets_); in Start()
594 s = SetOutputs(kernel_, ctx_, args_); in Finish()
641 return SetOutputs(this, ctx, args); in DoComputeSync()
841 s = SetOutputs(kernel_, ctx_, rets_); in Finish()
|
| /external/tensorflow/tensorflow/lite/examples/label_image/ |
| D | bitmap_helpers_impl.h | 44 interpreter->SetOutputs({2}); in resize()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 196 EXPECT_EQ(interpreter_.SetOutputs({4}), kTfLiteOk); in InterpreterFp16()
469 EXPECT_EQ(interpreter_.SetOutputs({3}), kTfLiteOk); in InterpreterFp32()
584 EXPECT_EQ(interpreter_.SetOutputs({7}), kTfLiteOk); in Interpreter2Fp32()
764 EXPECT_EQ(interpreter_.SetOutputs({6, 7}), kTfLiteOk); in InterpreterMultiNode()
1084 EXPECT_EQ(interpreter_.SetOutputs({5}), kTfLiteOk); in InterpreterQuantized()
|
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate_test.cc | 33 interpreter_->SetOutputs({3, 4}); in SetUp()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test_util.cc | 100 subgraph->SetOutputs({3, 4}); in SetUpSubgraph()
335 interpreter_->SetOutputs({12}); in SetUp()
|
| D | delegate_test.cc | 814 interpreter_->SetOutputs({1, 2}); in SetUp()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | test_util.cc | 92 CHECK_EQ(interpreter_->SetOutputs(outputs), kTfLiteOk); in AddTensors()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 68 TfLiteStatus SetOutputs(std::vector<int> outputs);
|
| D | subgraph.cc | 546 TfLiteStatus Subgraph::SetOutputs(std::vector<int> outputs) { in SetOutputs() function in tflite::Subgraph
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops_test.cc | 340 ASSERT_EQ(interpreter_->SetOutputs({kResultTensorId, kSizeTensorId}), in AddTensors()
352 interpreter_->SetOutputs({kResultTensorId, kSizeTensorId, in AddTensors()
|