Home
last modified time | relevance | path

Searched refs:AddOutput (Results 1 – 25 of 221) sorted by relevance

123456789

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Darithmetic_builder.cc65 const auto& math_out = AddOutput(sizeof(int), 4, in PopulateSubGraph()
68 const auto& math_out_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
69 const auto& math_out_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
79 requantize_op->AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
82 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
83 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
85 auto result_out = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
88 auto result_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
89 auto result_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
99 requantize_op->AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
[all …]
Dpool_2d_builder.cc67 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
70 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
71 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
75 TensorID pool_out = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
78 const auto& pool_out_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
79 const auto& pool_out_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
97 requantize_op->AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
100 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
101 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dconv_2d_builder.cc265 AddOutput(sizeof(uint8_t), 4, space_to_batch_output_dims); in PopulateSubGraph()
266 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
267 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
303 TensorID conv_output = conv_op->AddOutput( in PopulateSubGraph()
308 conv_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
309 conv_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
320 batch_to_space_op->AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
324 batch_to_space_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
326 batch_to_space_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
352 output_tensor = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
[all …]
Dreduce_builder.cc72 auto mean_output = AddOutput(output_element_size, 4, in PopulateSubGraph()
75 auto mean_out_min = AddOutput(output_element_size, 4, kScalarShape); in PopulateSubGraph()
76 auto mean_out_max = AddOutput(output_element_size, 4, kScalarShape); in PopulateSubGraph()
87 requantize_op->AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
90 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
91 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Drsqrt_builder.cc101 auto sqrt_output = AddOutput( in PopulateSubGraph()
103 auto sqrt_output_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
104 auto sqrt_output_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
116 auto reshape_out = reshape_op->AddOutput(sizeof(uint8_t), 4, flat_shape); in PopulateSubGraph()
147 auto div_output = div_op->AddOutput(sizeof(uint8_t), 4, flat_shape); in PopulateSubGraph()
148 div_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
149 div_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
162 node_output_ = reshape_op->AddOutput(sizeof(uint8_t), 4, output_shape); in PopulateSubGraph()
Dmatmul_builder.cc79 matmul_op->AddOutput(sizeof(int), 4, in AddFullyConnectedHelper()
83 matmul_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
85 matmul_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
111 bias_add_op->AddOutput(sizeof(int), 4, in AddFullyConnectedHelper()
115 bias_add_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
117 bias_add_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
139 quantize_biasadd_op->AddOutput(sizeof(uint8_t), 4, in AddFullyConnectedHelper()
142 quantize_biasadd_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
143 quantize_biasadd_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
251 auto transposed_weights = AddOutput(sizeof(uint8_t), 4, weights_shape_); in PopulateSubGraph()
[all …]
Dsquared_difference.cc64 auto sub_out = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
67 auto sub_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
68 auto sub_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
81 node_output_ = mul_op->AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
84 mul_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
85 mul_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dpack_builder.cc86 TensorID pack_out = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
99 const auto& pack_out_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
100 const auto& pack_out_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
111 requantize_op->AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
114 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
115 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dconcat_builder.cc89 TensorID concat_out = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
92 const auto& concat_out_min = AddOutput(sizeof(float), 4, {1, 1, 1, 1}); in PopulateSubGraph()
93 const auto& concat_out_max = AddOutput(sizeof(float), 4, {1, 1, 1, 1}); in PopulateSubGraph()
118 requantize_op->AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
121 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
122 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dneg_op_builder.cc36 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
39 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
40 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dl2_normalization_builder.cc42 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
45 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
46 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dtranspose_builder.cc41 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
44 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
45 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dquantize_builder.cc43 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
46 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
47 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dhardswish_builder.cc45 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
48 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
49 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dspace_to_depth_builder.cc52 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
55 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
56 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dactivation_builder.cc50 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
53 AddOutput(sizeof(float), 4, {1, 1, 1, 1}); in PopulateSubGraph()
54 AddOutput(sizeof(float), 4, {1, 1, 1, 1}); in PopulateSubGraph()
Dsoftmax_builder.cc51 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
54 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
55 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dpad_builder.cc57 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
60 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
61 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dmin_max_builder.cc47 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
50 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
51 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dcast_builder.cc61 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
63 AddOutput(sizeof(float), 4, {1, 1, 1, 1}); in PopulateSubGraph()
64 AddOutput(sizeof(float), 4, {1, 1, 1, 1}); in PopulateSubGraph()
Dsplit_builder.cc64 TensorID output = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
70 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
71 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dactivation_layer_builder.cc67 builder_output_ = AddOutput(); in PopulateSubgraph()
81 threshold_builder->AddInput(AddOutput()); in PopulateSubgraph()
89 negation_builder->AddInput(threshold_builder->AddOutput()); in PopulateSubgraph()
90 builder_output_ = negation_builder->AddOutput(); in PopulateSubgraph()
/external/tensorflow/tensorflow/lite/micro/kernels/
Delu_test.cc32 output_ = AddOutput({input.type, {}, 0, 0, 1. / 256}); in BaseActivationsOpModel()
34 output_ = AddOutput({input.type, {}, 0, 0, 1. / 256, -128}); in BaseActivationsOpModel()
36 output_ = AddOutput({input.type, {}}); in BaseActivationsOpModel()
/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression_test.cc85 output_selected_indices_ = AddOutput(TensorType_INT32); in NonMaxSuppressionV4OpModel()
87 output_num_selected_indices_ = AddOutput(TensorType_INT32); in NonMaxSuppressionV4OpModel()
177 output_selected_indices_ = AddOutput(TensorType_INT32); in NonMaxSuppressionV5OpModel()
178 output_selected_scores_ = AddOutput(TensorType_FLOAT32); in NonMaxSuppressionV5OpModel()
179 output_num_selected_indices_ = AddOutput(TensorType_INT32); in NonMaxSuppressionV5OpModel()
Dquant_basic_lstm_test.cc77 sum_out_ = AddOutput(input_tensor_data); in QuantizedLSTMOpModel()
80 AddOutput({TensorType_UINT8, output_shape, 0.0f, 0.0f, 1. / 128., 128}); in QuantizedLSTMOpModel()
82 AddOutput({TensorType_INT16, state_shape, 0.0f, 0.0f, 1. / 2048., 0}); in QuantizedLSTMOpModel()
84 AddOutput({TensorType_UINT8, output_shape, 0.0f, 0.0f, 1. / 128., 128}); in QuantizedLSTMOpModel()
86 AddOutput({TensorType_INT16, output_shape, 0.0f, 0.0f, 1. / 128., 128}); in QuantizedLSTMOpModel()

123456789