Searched refs:SetupTensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 75 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor() function
105 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildAddSubgraph()
106 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildAddSubgraph()
107 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildAddSubgraph()
136 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildMulSubgraph()
137 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildMulSubgraph()
138 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildMulSubgraph()
167 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildPadSubgraph()
168 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildPadSubgraph()
169 SetupTensor(subgraph, kOutput, kTfLiteInt32); in BuildPadSubgraph()
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| D | call_test.cc | 39 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor() function in tflite::__anon5594bfc90111::CallTest
57 SetupTensor(subgraph, idx, kTfLiteInt32); in BuildCallSubgraph()
60 SetupTensor(subgraph, idx, kTfLiteInt32); in BuildCallSubgraph()
105 SetupTensor(subgraph, kInput1, kTfLiteInt32); in BuildGraphWithMultipleOutputs()
106 SetupTensor(subgraph, kInput2, kTfLiteInt32); in BuildGraphWithMultipleOutputs()
107 SetupTensor(subgraph, kMulOutput, kTfLiteInt32); in BuildGraphWithMultipleOutputs()
108 SetupTensor(subgraph, kAddOutput, kTfLiteInt32); in BuildGraphWithMultipleOutputs()
147 SetupTensor(this_subgraph, idx, kTfLiteInt32); in BuildMultiNodeGraph()
|