Home
last modified time | relevance | path

Searched defs:OpData (Results 1 – 25 of 61) sorted by relevance

123

/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Ddecode_jpeg_register.h39 struct OpData { struct
41 int32_t num_images;
44 int32_t height;
46 int32_t width;
48 int32_t channels;
/external/tensorflow/tensorflow/lite/kernels/
Dadd.cc54 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 …]
Dsub.cc56 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 …]
Dconv.cc73 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 …]
Dtranspose_conv.cc56 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 …]
Ddepthwise_conv.cc61 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 …]
Dconv3d.cc46 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;
Dsvdf.cc40 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;
Ddiv.cc48 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;
Dnumeric_verify.cc58 struct OpData { struct
60 float tolerance;
62 bool float_input_initialized;
63 int cache_tensor_id = kTensorNotAllocated;
65 bool log_if_failed;
Dmul.cc54 struct OpData { struct
56 int32 output_activation_min;
57 int32 output_activation_max;
60 int32_t output_multiplier;
61 int output_shift;
Dwhile.cc32 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;
Dconv3d_transpose.cc41 struct OpData { struct
42 Padding3DValues padding;
45 int col2im_id = kTensorNotAllocated;
48 int col2im_index;
50 bool need_col2im = false;
Dbucketize.cc35 struct OpData { struct
37 const float* boundaries;
38 int num_boundaries;
Dselect.cc39 struct OpData { struct
40 bool requires_broadcast;
43 bool has_low_rank_input_condition;
Droll.cc100 struct OpData { struct
103 int cache_tensor_id = kTensorNotAllocated;
104 int32_t cache_index = kTensorNotAllocated;
105 bool need_cache = false;
Ddensify.cc39 struct OpData { struct
40 bool dense_weights_initialized;
Delementwise.cc41 struct OpData { struct
42 int32_t multiplier;
43 int32_t shift;
44 int input_offset;
45 int output_offset;
46 bool needs_rescale;
Dquantize.cc41 struct OpData { struct
42 int32_t output_multiplier;
43 int output_shift;
Dif.cc33 struct OpData { struct
34 int then_subgraph_index;
35 int else_subgraph_index;
Ddetection_postprocess.cc89 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 …]
Dcall_once.cc40 struct OpData { struct
42 int init_subgraph_index;
/external/llvm/utils/TableGen/
DPseudoLoweringEmitter.cpp27 struct OpData { struct in __anon32c82d640111::PseudoLoweringEmitter
28 enum MapKind { Operand, Imm, Reg };
29 MapKind Kind;
30 union {
34 } Data;
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example.cc653 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 …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc392 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;

123