Searched refs:SetupTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 46 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor() function
76 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildAddSubgraph()
77 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildAddSubgraph()
78 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildAddSubgraph()
106 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildMulSubgraph()
107 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildMulSubgraph()
108 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildMulSubgraph()
136 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildPadSubgraph()
137 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildPadSubgraph()
138 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildPadSubgraph()
[all …]
|