| /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 | 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 | 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 | 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 | depth_to_space_tester.cc | 149 const std::array<int32_t, 1> subgraph_outputs{{op_outputs.front()}}; in CreateTfLiteModel() local
154 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
155 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 | 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 | 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 | prelu_tester.cc | 211 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
217 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
218 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | fully_connected_tester.cc | 261 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
267 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
268 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | conv_2d_tester.cc | 320 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
326 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
327 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | depthwise_conv_2d_tester.cc | 312 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
318 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
319 subgraph_outputs.size()), in CreateTfLiteModel()
|
| D | binary_elementwise_tester.cc | 384 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
390 builder.CreateVector<int32_t>(subgraph_outputs.data(), in CreateTfLiteModel()
391 subgraph_outputs.size()), in CreateTfLiteModel()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 135 void InitializeGraphOutputs(const std::set<int>& subgraph_outputs) { in InitializeGraphOutputs() argument
138 subgraph_outputs_.push_back(subgraph_outputs.count(i) > 0); in InitializeGraphOutputs()
372 std::vector<int> subgraph_outputs; member
388 op_data_->subgraph_outputs.push_back(tensor_index); in Init()
482 for (auto tensor_index : op_data_->subgraph_outputs) { in Prepare()
621 for (auto tensor_index : op_data_->subgraph_outputs) { in Eval()
|
| /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()
|