| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | leaky_relu_tester.cc | 130 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
135 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
136 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | softmax_tester.cc | 132 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
137 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
138 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | reduce_tester.cc | 147 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
152 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
153 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | unary_elementwise_tester.cc | 157 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
162 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
163 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | reshape_tester.cc | 160 const std::array<int32_t, 1> subgraph_outputs{{op_outputs.front()}}; in CreateTfLiteModel() local
165 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
166 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | dequantize_tester.cc | 141 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
146 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
147 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | transpose_tester.cc | 160 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
165 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
166 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_leaky_relu_tester.cc | 152 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
157 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
158 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_unary_elementwise_tester.cc | 150 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
155 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
156 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | resize_bilinear_tester.cc | 164 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
169 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
170 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_reduce_tester.cc | 166 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
171 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
172 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | pool_2d_tester.cc | 177 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
182 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
183 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | pad_tester.cc | 163 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
168 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
169 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_pad_tester.cc | 181 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
186 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
187 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_pool_2d_tester.cc | 183 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
188 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
189 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_resize_bilinear_tester.cc | 182 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
187 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
188 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantize_tester.cc | 203 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
208 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
209 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | concatenation_tester.cc | 211 const std::array<int32_t, 1> subgraph_outputs = op_outputs; in CreateTfLiteModel() local
216 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
217 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | depth_to_space_tester.cc | 228 const std::array<int32_t, 1> subgraph_outputs{{op_outputs.front()}}; in CreateTfLiteModel() local
233 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
234 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_fully_connected_tester.cc | 220 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
226 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
227 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | split_tester.cc | 216 const std::vector<int32_t> subgraph_outputs = op_outputs; in CreateTfLiteModel() local
221 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
222 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_conv_2d_tester.cc | 252 const std::array<int32_t, 1> subgraph_outputs{{3}}; in CreateTfLiteModel() local
257 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
258 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | quantized_depthwise_conv_2d_tester.cc | 257 const std::array<int32_t, 1> subgraph_outputs{{3}}; in CreateTfLiteModel() local
262 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
263 subgraph_outputs.size()), in CreateTfLiteModel()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/ |
| D | add.cc | 67 int32_t subgraph_outputs[1] = {2}; in Build() local
71 builder.CreateVector<int32_t>(subgraph_outputs, 1), in Build()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 140 void InitializeGraphOutputs(const std::set<int>& subgraph_outputs) { in InitializeGraphOutputs() argument
143 subgraph_outputs_.push_back(subgraph_outputs.count(i) > 0); in InitializeGraphOutputs()
397 std::vector<int> subgraph_outputs; member
456 op_data_->subgraph_outputs.push_back(tensor_index); in Init()
583 for (auto tensor_index : op_data_->subgraph_outputs) { in Prepare()
762 for (auto tensor_index : op_data_->subgraph_outputs) { in Eval()
|