| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | arithmetic_builder.cc | 65 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 …]
|
| D | pool_2d_builder.cc | 67 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()
|
| D | conv_2d_builder.cc | 265 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 …]
|
| D | reduce_builder.cc | 72 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()
|
| D | rsqrt_builder.cc | 101 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()
|
| D | squared_difference.cc | 64 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()
|
| D | matmul_builder.cc | 79 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 …]
|
| D | pack_builder.cc | 86 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()
|
| D | concat_builder.cc | 89 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()
|
| D | l2_normalization_builder.cc | 42 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()
|
| D | neg_op_builder.cc | 36 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()
|
| D | transpose_builder.cc | 41 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()
|
| D | hardswish_builder.cc | 45 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()
|
| D | quantize_builder.cc | 43 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()
|
| D | space_to_depth_builder.cc | 52 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()
|
| D | activation_builder.cc | 50 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()
|
| D | pad_builder.cc | 57 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()
|
| D | min_max_builder.cc | 47 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()
|
| D | softmax_builder.cc | 51 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()
|
| D | cast_builder.cc | 61 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()
|
| D | resize_nearest_neighbor_builder.cc | 71 node_output_ = AddOutput(sizeof(uint8_t), 4, in PopulateSubGraph()
74 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
75 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
|
| D | split_builder.cc | 64 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/micro/kernels/ |
| D | elu_test.cc | 32 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/delegates/coreml/builders/ |
| D | activation_layer_builder.cc | 67 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/kernels/ |
| D | non_max_suppression_test.cc | 85 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()
|