Home
last modified time | relevance | path

Searched defs:subgraph_outputs (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc67 int32_t subgraph_outputs[1] = {2}; in Build() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dleaky_relu_tester.cc130 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
Dsoftmax_tester.cc132 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
Dreshape_tester.cc160 const std::array<int32_t, 1> subgraph_outputs{{op_outputs.front()}}; in CreateTfLiteModel() local
Ddepth_to_space_tester.cc149 const std::array<int32_t, 1> subgraph_outputs{{op_outputs.front()}}; in CreateTfLiteModel() local
Dunary_elementwise_tester.cc157 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
Dreduce_tester.cc147 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
Dresize_bilinear_tester.cc164 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
Dpad_tester.cc163 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
Dprelu_tester.cc211 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
Dpool_2d_tester.cc177 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
Dfully_connected_tester.cc261 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
Dbinary_elementwise_tester.cc384 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
Dconv_2d_tester.cc320 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
Ddepthwise_conv_2d_tester.cc312 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc135 void InitializeGraphOutputs(const std::set<int>& subgraph_outputs) { in InitializeGraphOutputs()
372 std::vector<int> subgraph_outputs; member