Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel.cc140 void InitializeGraphOutputs(const std::set<int>& subgraph_outputs) { in InitializeGraphOutputs()
397 std::vector<int> subgraph_outputs; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
H A Dadd.cc67 int32_t subgraph_outputs[1] = {2}; in Build() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dsoftmax_tester.cc132 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Dleaky_relu_tester.cc130 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Dreshape_tester.cc160 const std::array<int32_t, 1> subgraph_outputs{{op_outputs.front()}}; in CreateTfLiteModel() local
H A Dunary_elementwise_tester.cc157 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Ddequantize_tester.cc141 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Dtranspose_tester.cc160 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
H A Dquantized_leaky_relu_tester.cc152 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Dquantized_unary_elementwise_tester.cc150 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Dreduce_tester.cc147 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
H A Dresize_bilinear_tester.cc164 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
H A Dpad_tester.cc163 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
H A Dquantized_resize_bilinear_tester.cc182 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
H A Dquantized_reduce_tester.cc166 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
H A Dquantized_pool_2d_tester.cc183 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Dquantized_pad_tester.cc181 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
H A Dpool_2d_tester.cc177 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Dquantized_fully_connected_tester.cc220 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
H A Dprelu_tester.cc268 const std::array<int32_t, 1> subgraph_outputs{ in CreateTfLiteModel() local
H A Dquantize_tester.cc203 const std::array<int32_t, 1> subgraph_outputs{{1}}; in CreateTfLiteModel() local
H A Dquantized_depthwise_conv_2d_tester.cc257 const std::array<int32_t, 1> subgraph_outputs{{3}}; in CreateTfLiteModel() local
H A Dquantized_binary_elementwise_tester.cc255 const std::array<int32_t, 1> subgraph_outputs{{2}}; in CreateTfLiteModel() local
H A Dquantized_conv_2d_tester.cc252 const std::array<int32_t, 1> subgraph_outputs{{3}}; in CreateTfLiteModel() local
H A Dsplit_tester.cc216 const std::vector<int32_t> subgraph_outputs = op_outputs; in CreateTfLiteModel() local

12