Searched refs:kInput2 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 62 const int kInput2 = 1; in BuildAddSubgraph() local
73 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildAddSubgraph()
77 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildAddSubgraph()
85 {kInput1, kInput2}, {kOutput}, nullptr, 0, params, in BuildAddSubgraph()
92 const int kInput2 = 1; in BuildMulSubgraph() local
103 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildMulSubgraph()
107 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildMulSubgraph()
115 {kInput1, kInput2}, {kOutput}, nullptr, 0, params, in BuildMulSubgraph()
122 const int kInput2 = 1; in BuildPadSubgraph() local
133 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildPadSubgraph()
[all …]
|