| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | conv.cc | 73 struct OpData { struct
76 int im2col_id = kTensorNotAllocated;
77 int hwcn_weights_id = kTensorNotAllocated;
78 int input_quantized_id = kTensorNotAllocated;
79 int scaling_factors_id = kTensorNotAllocated;
80 int input_offset_id = kTensorNotAllocated;
81 int accum_scratch_id = kTensorNotAllocated;
83 int row_sums_id = kTensorNotAllocated;
85 TfLitePaddingValues padding;
88 int32_t output_multiplier;
[all …]
|
| H A D | add.cc | 54 struct OpData { struct
57 int input1_shift;
58 int input2_shift;
59 int32 output_activation_min;
60 int32 output_activation_max;
63 int32 input1_multiplier;
64 int32 input2_multiplier;
65 int32 output_multiplier;
66 int output_shift;
67 int left_shift;
[all …]
|
| H A D | sub.cc | 56 struct OpData { struct
57 bool requires_broadcast;
61 int input1_shift;
62 int input2_shift;
63 int32 output_activation_min;
64 int32 output_activation_max;
67 int32 input1_multiplier;
68 int32 input2_multiplier;
69 int32 output_multiplier;
70 int output_shift;
[all …]
|
| H A D | transpose_conv.cc | 56 struct OpData { struct
59 int col2im_id = kTensorNotAllocated;
60 int transposed_weights_id = kTensorNotAllocated;
61 int scratch_tensor_id = kTensorNotAllocated;
65 int32_t col2im_index;
70 int32_t transposed_weights_index;
74 int32_t scratch_tensor_index;
76 TfLitePaddingValues padding;
79 int32_t output_multiplier;
80 int output_shift;
[all …]
|
| H A D | depthwise_conv.cc | 61 struct OpData { struct
62 TfLitePaddingValues padding;
65 int32_t output_multiplier;
66 int output_shift;
69 int32_t output_activation_min;
70 int32_t output_activation_max;
73 std::vector<int32_t> per_channel_output_multiplier;
74 std::vector<int> per_channel_output_shift;
77 int input_quantized_id = kTensorNotAllocated;
78 int scaling_factors_id = kTensorNotAllocated;
[all …]
|
| H A D | conv3d.cc | 46 struct OpData { struct
47 Padding3DValues padding;
48 int im2col_tensor_id = kTensorNotAllocated;
49 int transposed_filter_tensor_id = kTensorNotAllocated;
51 bool need_im2col = false;
52 bool need_transposed_filter = false;
56 bool im2col_oversized = false;
58 int32_t im2col_index;
59 int32_t transposed_filter_index;
|
| H A D | svdf.cc | 40 struct OpData { struct
41 int scratch_tensor_index;
42 bool float_weights_time_initialized;
43 int32 effective_scale_1_a;
44 int effective_scale_1_b;
45 int32 effective_scale_2_a;
46 int effective_scale_2_b;
47 bool compute_row_sums = false;
|
| H A D | div.cc | 48 struct OpData { struct
49 bool requires_broadcast;
52 int32 output_activation_min;
53 int32 output_activation_max;
56 int32_t output_multiplier;
57 int output_shift;
|
| H A D | while.cc | 32 struct OpData { struct
33 int cond_subgraph_index;
34 int body_subgraph_index;
35 bool cond_has_dynamic_output_tensors;
36 bool body_has_dynamic_output_tensors;
37 bool body_use_shallow_copy;
39 bool subgraphs_prepared;
|
| H A D | detection_postprocess.cc | 89 struct OpData { struct
90 int max_detections;
91 int max_classes_per_detection; // Fast Non-Max-Suppression
92 int detections_per_class; // Regular Non-Max-Suppression
93 float non_max_suppression_score_threshold;
94 float intersection_over_union_threshold;
95 int num_classes;
96 bool use_regular_non_max_suppression;
97 CenterSizeEncoding scale_values;
99 int decoded_boxes_index;
[all …]
|
| H A D | numeric_verify.cc | 58 struct OpData { struct
60 float tolerance;
62 bool float_input_initialized;
63 int cache_tensor_id = kTensorNotAllocated;
65 bool log_if_failed;
|
| H A D | mul.cc | 54 struct OpData { struct
56 int32 output_activation_min;
57 int32 output_activation_max;
60 int32_t output_multiplier;
61 int output_shift;
|
| H A D | batch_matmul.cc | 55 struct OpData { struct
58 int32_t output_multiplier;
59 int output_shift;
62 int32_t output_activation_min;
63 int32_t output_activation_max;
65 int scratch_tensor_index;
66 bool rhs_transposed;
67 bool compute_row_sums = false;
|
| H A D | fully_connected.cc | 107 struct OpData { struct
110 int32_t output_multiplier;
111 int output_shift;
113 std::vector<int32_t> per_channel_output_multiplier;
114 std::vector<int> per_channel_output_shift;
117 int32_t output_activation_min;
118 int32_t output_activation_max;
120 int scratch_tensor_index;
121 bool compute_row_sums = false;
123 bool ledger_initialized;
|
| H A D | conv3d_transpose.cc | 41 struct OpData { struct
42 Padding3DValues padding;
45 int col2im_id = kTensorNotAllocated;
48 int col2im_index;
50 bool need_col2im = false;
|
| H A D | elementwise.cc | 41 struct OpData { struct
42 int32_t multiplier;
43 int32_t shift;
44 int input_offset;
45 int output_offset;
46 bool needs_rescale;
|
| H A D | roll.cc | 100 struct OpData { struct
103 int cache_tensor_id = kTensorNotAllocated;
104 int32_t cache_index = kTensorNotAllocated;
105 bool need_cache = false;
|
| H A D | bucketize.cc | 35 struct OpData { struct
37 const float* boundaries;
38 int num_boundaries;
|
| H A D | select.cc | 39 struct OpData { struct
40 bool requires_broadcast;
43 bool has_low_rank_input_condition;
|
| H A D | if.cc | 33 struct OpData { struct
34 int then_subgraph_index;
35 int else_subgraph_index;
|
| H A D | quantize.cc | 41 struct OpData { struct
42 int32_t output_multiplier;
43 int output_shift;
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| H A D | decode_jpeg_register.h | 39 struct OpData { struct
41 int32_t num_images;
44 int32_t height;
46 int32_t width;
48 int32_t channels;
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| H A D | parse_example.cc | 653 struct OpData { struct
654 FastParseExampleConfig config;
655 std::vector<tf::TensorShape> dense_shapes;
656 int dense_size = 0;
657 int sparse_size = 0;
658 std::unique_ptr<ConfigIndex> config_index;
659 int config_index_size;
660 SeededHasher hasher;
661 TfLiteResult got;
662 bool* quick_filter = nullptr;
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | kernel.cc | 392 struct OpData { struct
393 tensorflow::EagerContext* eager_context;
394 tensorflow::CancellationManager* cancellation_manager;
395 std::vector<std::unique_ptr<OpNode>> nodes;
396 std::vector<int> subgraph_inputs;
397 std::vector<int> subgraph_outputs;
399 disable_reusing_buffer_tensors; // A list of input tensor indexes which
402 OpDataInfo shared_info;
|
| /aosp_15_r20/external/llvm/utils/TableGen/ |
| H A D | PseudoLoweringEmitter.cpp | 27 struct OpData { struct in __anone8c432e40111::PseudoLoweringEmitter
28 enum MapKind { Operand, Imm, Reg };
29 MapKind Kind;
30 union {
34 } Data;
|