Home
last modified time | relevance | path

Searched refs:op_outputs (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dreshape_tester.cc140 const std::array<int32_t, 1> op_outputs{{OutputShapeAsInput() ? 2 : 1}}; in CreateTfLiteModel() local
156 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
160 const std::array<int32_t, 1> subgraph_outputs{{op_outputs.front()}}; in CreateTfLiteModel()
Ddepth_to_space_tester.cc140 const std::array<int32_t, 1> op_outputs{{1}}; in CreateTfLiteModel() local
144 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
149 const std::array<int32_t, 1> subgraph_outputs{{op_outputs.front()}}; in CreateTfLiteModel()
Dleaky_relu_tester.cc122 const std::array<int32_t, 1> op_outputs{{1}}; in CreateTfLiteModel() local
126 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
Dsoftmax_tester.cc124 const std::array<int32_t, 1> op_outputs{{1}}; in CreateTfLiteModel() local
128 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
Dunary_elementwise_tester.cc150 const std::array<int32_t, 1> op_outputs{{1}}; in CreateTfLiteModel() local
154 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in CreateTfLiteModel()
Dreduce_tester.cc139 const std::array<int32_t, 1> op_outputs{{2}}; in CreateTfLiteModel() local
143 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
Dresize_bilinear_tester.cc156 const std::array<int32_t, 1> op_outputs{{2}}; in CreateTfLiteModel() local
160 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
Dpad_tester.cc156 const std::array<int32_t, 1> op_outputs{{2}}; in CreateTfLiteModel() local
160 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in CreateTfLiteModel()
Dpool_2d_tester.cc168 const std::array<int32_t, 1> op_outputs{{1}}; in CreateTfLiteModel() local
173 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
Dprelu_tester.cc202 const std::array<int32_t, 1> op_outputs{ in CreateTfLiteModel() local
207 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()))); in CreateTfLiteModel()
Dfully_connected_tester.cc251 const std::array<int32_t, 1> op_outputs{ in CreateTfLiteModel() local
256 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
Dconv_2d_tester.cc306 const std::array<int32_t, 1> op_outputs{ in CreateTfLiteModel() local
315 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
Ddepthwise_conv_2d_tester.cc297 const std::array<int32_t, 1> op_outputs{ in CreateTfLiteModel() local
307 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size()), in CreateTfLiteModel()
/external/tensorflow/tensorflow/python/eager/
Dlift_to_graph.py252 op_outputs = collections.defaultdict(set)
261 op_outputs=op_outputs,
269 if not op_outputs[_as_operation(t)]]
284 if (all(x in marked_ops for x in op_outputs[inp]) and
/external/tensorflow/tensorflow/c/eager/
Dcustom_device_testutil.cc131 std::vector<TFE_TensorHandle*> op_outputs(*num_outputs); in LoggingDeviceExecute() local
132 TFE_Execute(op, op_outputs.data(), num_outputs, s); in LoggingDeviceExecute()
136 unwrapped_outputs.reserve(op_outputs.size()); in LoggingDeviceExecute()
137 for (auto* handle : op_outputs) { in LoggingDeviceExecute()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc58 const int32_t op_outputs[1] = {2}; in Build() local
63 builder.CreateVector<int32_t>(op_outputs, 1), in Build()
/external/XNNPACK/bench/
Dbankers-rounding.cc134 const std::array<int32_t, 1> op_outputs{{ 1 }}; in tflite_bankers_rounding_f32() local
139 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in tflite_bankers_rounding_f32()
Delu.cc134 const std::array<int32_t, 1> op_outputs{{ 1 }}; in tflite_elu_f32() local
139 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in tflite_elu_f32()
Dceiling.cc134 const std::array<int32_t, 1> op_outputs{{ 1 }}; in tflite_ceiling_f32() local
139 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in tflite_ceiling_f32()
Dsquare-root.cc134 const std::array<int32_t, 1> op_outputs{{ 1 }}; in tflite_square_root_f32() local
139 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in tflite_square_root_f32()
Dfloor.cc134 const std::array<int32_t, 1> op_outputs{{ 1 }}; in tflite_floor_f32() local
139 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in tflite_floor_f32()
Dsigmoid.cc209 const std::array<int32_t, 1> op_outputs{{ 1 }}; in tflite_sigmoid_f32() local
214 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in tflite_sigmoid_f32()
Dhardswish.cc136 const std::array<int32_t, 1> op_outputs{{ 1 }}; in tflite_hardswish_f32() local
141 builder.CreateVector<int32_t>(op_outputs.data(), op_outputs.size())); in tflite_hardswish_f32()
Dprelu.cc155 const int32_t op_outputs[1] = { 2 }; in tflite_prelu_f32() local
160 builder.CreateVector<int32_t>(op_outputs, 1)); in tflite_prelu_f32()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.cc371 auto op_outputs = op->outputs(); in BuildLoggingInterpreter() local
373 op_info.outputs = std::vector<int>(op_outputs->begin(), op_outputs->end()); in BuildLoggingInterpreter()

12