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