Home
last modified time | relevance | path

Searched refs:subgraph_inputs (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dleaky_relu_tester.cc129 const std::array<int32_t, 1> subgraph_inputs{{0}}; in CreateTfLiteModel() local
133 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
134 subgraph_inputs.size()), in CreateTfLiteModel()
Dsoftmax_tester.cc131 const std::array<int32_t, 1> subgraph_inputs{{0}}; in CreateTfLiteModel() local
135 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
136 subgraph_inputs.size()), in CreateTfLiteModel()
Dreshape_tester.cc159 const std::array<int32_t, 1> subgraph_inputs{{op_inputs.front()}}; in CreateTfLiteModel() local
163 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
164 subgraph_inputs.size()), in CreateTfLiteModel()
Dunary_elementwise_tester.cc156 const std::array<int32_t, 1> subgraph_inputs{{0}}; in CreateTfLiteModel() local
160 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
161 subgraph_inputs.size()), in CreateTfLiteModel()
Dreduce_tester.cc146 const std::array<int32_t, 1> subgraph_inputs{{0}}; in CreateTfLiteModel() local
150 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
151 subgraph_inputs.size()), in CreateTfLiteModel()
Ddepth_to_space_tester.cc148 const std::array<int32_t, 1> subgraph_inputs{{op_inputs.front()}}; in CreateTfLiteModel() local
152 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
153 subgraph_inputs.size()), in CreateTfLiteModel()
Dresize_bilinear_tester.cc163 const std::array<int32_t, 1> subgraph_inputs{{0}}; in CreateTfLiteModel() local
167 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
168 subgraph_inputs.size()), in CreateTfLiteModel()
Dpad_tester.cc162 const std::array<int32_t, 1> subgraph_inputs{{0}}; in CreateTfLiteModel() local
166 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
167 subgraph_inputs.size()), in CreateTfLiteModel()
Dpool_2d_tester.cc176 const std::array<int32_t, 1> subgraph_inputs{{0}}; in CreateTfLiteModel() local
180 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
181 subgraph_inputs.size()), in CreateTfLiteModel()
Dbinary_elementwise_tester.cc377 std::vector<int32_t> subgraph_inputs; in CreateTfLiteModel() local
379 subgraph_inputs.push_back(tensors.size() - 3); in CreateTfLiteModel()
382 subgraph_inputs.push_back(tensors.size() - 2); in CreateTfLiteModel()
388 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
389 subgraph_inputs.size()), in CreateTfLiteModel()
Dprelu_tester.cc209 const std::array<int32_t, 1> subgraph_inputs{ in CreateTfLiteModel() local
215 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
216 subgraph_inputs.size()), in CreateTfLiteModel()
Dfully_connected_tester.cc259 const std::array<int32_t, 1> subgraph_inputs{ in CreateTfLiteModel() local
265 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
266 subgraph_inputs.size()), in CreateTfLiteModel()
Dconv_2d_tester.cc318 const std::array<int32_t, 1> subgraph_inputs{ in CreateTfLiteModel() local
324 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
325 subgraph_inputs.size()), in CreateTfLiteModel()
Ddepthwise_conv_2d_tester.cc310 const std::array<int32_t, 1> subgraph_inputs{ in CreateTfLiteModel() local
316 builder.CreateVector<int32_t>(subgraph_inputs.data(), in CreateTfLiteModel()
317 subgraph_inputs.size()), in CreateTfLiteModel()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc66 int32_t subgraph_inputs[2] = {0, 1}; in Build() local
70 builder.CreateVector<int32_t>(subgraph_inputs, 2), in Build()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc371 std::vector<int> subgraph_inputs; member
394 op_data_->subgraph_inputs.push_back(tensor_index); in Init()
456 for (auto tensor_index : op_data_->subgraph_inputs) { in Prepare()
599 for (auto tensor_index : op_data_->subgraph_inputs) { in Eval()