Home
last modified time | relevance | path

Searched refs:op_data (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Ddetection_postprocess.cc120 OpData* op_data = nullptr; in Init() local
126 op_data = reinterpret_cast<OpData*>( in Init()
129 op_data->max_detections = m["max_detections"].AsInt32(); in Init()
130 op_data->max_classes_per_detection = m["max_classes_per_detection"].AsInt32(); in Init()
132 op_data->detections_per_class = kNumDetectionsPerClass; in Init()
134 op_data->detections_per_class = m["detections_per_class"].AsInt32(); in Init()
136 op_data->use_regular_non_max_suppression = false; in Init()
138 op_data->use_regular_non_max_suppression = m["use_regular_nms"].AsBool(); in Init()
140 op_data->non_max_suppression_score_threshold = in Init()
142 op_data->intersection_over_union_threshold = m["nms_iou_threshold"].AsFloat(); in Init()
[all …]
Dreduce.cc84 OpData* op_data = static_cast<OpData*>(node->user_data); in PrepareMax() local
89 op_data->input_scale = input->params.scale; in PrepareMax()
90 op_data->output_scale = output->params.scale; in PrepareMax()
91 op_data->num_output_elements = NumElements(output); in PrepareMax()
94 &op_data->temp_buffer_idx); in PrepareMax()
97 &op_data->resolved_axis_idx); in PrepareMax()
104 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in PrepareMeanOrSum() local
109 QuantizeMultiplier(real_multiplier, &op_data->multiplier, &op_data->shift); in PrepareMeanOrSum()
115 &op_data->temp_buffer_idx); in PrepareMeanOrSum()
116 op_data->input_zp = input->params.zero_point; in PrepareMeanOrSum()
[all …]
Dsoftmax_common.cc34 SoftmaxParams* op_data) { in CalculateSoftmaxParams() argument
67 QuantizeMultiplier(input_scale_beta_rescale, &op_data->input_multiplier, in CalculateSoftmaxParams()
69 op_data->input_left_shift = input_left_shift; in CalculateSoftmaxParams()
75 &op_data->input_multiplier, &input_left_shift); in CalculateSoftmaxParams()
76 op_data->input_left_shift = input_left_shift; in CalculateSoftmaxParams()
77 op_data->diff_min = in CalculateSoftmaxParams()
79 op_data->input_left_shift); in CalculateSoftmaxParams()
84 op_data->beta = static_cast<double>(params->beta); in CalculateSoftmaxParams()
106 SoftmaxParams* op_data = static_cast<SoftmaxParams*>(node->user_data); in SoftmaxPrepare() local
112 op_data->exp_lut = reinterpret_cast<int16_t*>(raw_exp_lut); in SoftmaxPrepare()
[all …]
Dfully_connected_common.cc35 const OpDataFullyConnected& op_data) { in FullyConnectedParamsQuantized() argument
37 op_params.input_offset = -op_data.input_zero_point; in FullyConnectedParamsQuantized()
38 op_params.weights_offset = -op_data.filter_zero_point; in FullyConnectedParamsQuantized()
39 op_params.output_offset = op_data.output_zero_point; in FullyConnectedParamsQuantized()
40 op_params.output_multiplier = op_data.output_multiplier; in FullyConnectedParamsQuantized()
41 op_params.output_shift = op_data.output_shift; in FullyConnectedParamsQuantized()
42 op_params.quantized_activation_min = op_data.output_activation_min; in FullyConnectedParamsQuantized()
43 op_params.quantized_activation_max = op_data.output_activation_max; in FullyConnectedParamsQuantized()
Dsoftmax.cc32 const SoftmaxParams& op_data) { in SoftmaxQuantized() argument
35 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
42 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
48 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
55 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
67 SoftmaxParams op_data = *static_cast<SoftmaxParams*>(node->user_data); in SoftmaxEval() local
72 op_data, tflite::micro::GetTensorShape(input), in SoftmaxEval()
81 SoftmaxQuantized(input, output, op_data); in SoftmaxEval()
Dcircular_buffer.cc80 OpData* op_data = static_cast<OpData*>(node->user_data); in Prepare() local
109 op_data->cycles_max = 1; in Prepare()
112 op_data->cycles_max = 2; in Prepare()
114 op_data->cycles_until_run = op_data->cycles_max; in Prepare()
115 node->user_data = op_data; in Prepare()
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_pool_with_argmax.cc117 OpData* op_data = new OpData; in Init() local
118 op_data->params.computed.padding = TfLitePaddingValues{0, 0, 0, 0}; in Init()
119 op_data->include_batch_in_index = m[kIncludeBatchStr].AsBool(); in Init()
120 op_data->params.activation = kTfLiteActNone; in Init()
124 op_data->params.padding = kTfLitePaddingValid; in Init()
126 op_data->params.padding = kTfLitePaddingSame; in Init()
128 op_data->params.padding = kTfLitePaddingUnknown; in Init()
136 op_data->params.filter_height = pool_size[1].AsInt32(); in Init()
137 op_data->params.filter_width = pool_size[2].AsInt32(); in Init()
144 op_data->params.stride_height = strides[1].AsInt32(); in Init()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dquantize.cc118 auto* op_data = static_cast<OpData*>(node->user_data); in EvalXtensa() local
120 auto* op_data = static_cast<OpDataQuantizeReference*>(node->user_data); in EvalXtensa() local
128 AffineQuantize(op_data->scale_multiplier, op_data->zero_point, in EvalXtensa()
140 op_data->input_zero_point, op_data->quantization_params.zero_point, in EvalXtensa()
141 op_data->requantize_output_shift, in EvalXtensa()
142 op_data->requantize_output_multiplier, size), in EvalXtensa()
154 int32_t zero_point = op_data->zero_point; in EvalXtensa()
156 int32_t zero_point = op_data->quantization_params.zero_point; in EvalXtensa()
159 size, op_data->requantize_output_multiplier, in EvalXtensa()
160 op_data->requantize_output_shift, in EvalXtensa()
[all …]
Dsoftmax.cc62 TfLiteStatus SoftmaxHifimini(OpData op_data, const RuntimeShape& input_shape, in SoftmaxHifimini() argument
86 input_diff == 0 ? kMaxExponentValue : op_data.exp_lut[input_diff]; in SoftmaxHifimini()
97 input_diff == 0 ? kMaxExponentValue : op_data.exp_lut[input_diff]; in SoftmaxHifimini()
120 OpData* op_data) { in CalculateSoftmaxOpDataHifimini() argument
151 op_data->exp_lut[i] = static_cast<uint16_t>(exponent_scaled); in CalculateSoftmaxOpDataHifimini()
167 OpData* op_data = static_cast<OpData*>(node->user_data); in PrepareHifimini() local
173 op_data->exp_lut = static_cast<uint16_t*>(context->AllocatePersistentBuffer( in PrepareHifimini()
175 TF_LITE_ENSURE(context, op_data->exp_lut != nullptr); in PrepareHifimini()
178 CalculateSoftmaxOpDataHifimini(context, input, output, params, op_data)); in PrepareHifimini()
212 TfLiteStatus EvalHifi4(const OpData* op_data, const TfLiteEvalTensor* input, in EvalHifi4() argument
[all …]
Dconv.cc247 const auto& op_data = *(reinterpret_cast<OpDataConv*>(node->user_data)); in Eval() local
267 -op_data.input_zero_point, op_data.output_zero_point, in Eval()
268 op_data.output_activation_min, op_data.output_activation_max, in Eval()
269 op_data.per_channel_output_multiplier, op_data.per_channel_output_shift, in Eval()
285 ConvPerChannel(ConvParamsQuantized(params, op_data), in Eval()
286 op_data.per_channel_output_multiplier, in Eval()
287 op_data.per_channel_output_shift, in Eval()
298 ConvParamsQuantized(params, op_data), in Eval()
299 op_data.per_channel_output_multiplier, in Eval()
300 op_data.per_channel_output_shift, in Eval()
/external/tensorflow/tensorflow/lite/kernels/
Dcall_once.cc48 auto* op_data = new OpData; in Init() local
50 op_data->init_subgraph_index = params->init_subgraph_index; in Init()
51 op_data->init_subgraph_invoked = false; in Init()
52 return op_data; in Init()
60 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
63 if (op_data->init_subgraph_invoked) return kTfLiteOk; in Prepare()
70 TF_LITE_ENSURE(context, op_data->init_subgraph_index < subgraphs->size()); in Prepare()
73 Subgraph* init_subgraph = (*subgraphs)[op_data->init_subgraph_index].get(); in Prepare()
80 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
83 if (op_data->init_subgraph_invoked) return kTfLiteOk; in Eval()
[all …]
Ddetection_postprocess.cc102 auto* op_data = new OpData; in Init() local
105 op_data->max_detections = m["max_detections"].AsInt32(); in Init()
106 op_data->max_classes_per_detection = m["max_classes_per_detection"].AsInt32(); in Init()
108 op_data->detections_per_class = kNumDetectionsPerClass; in Init()
110 op_data->detections_per_class = m["detections_per_class"].AsInt32(); in Init()
112 op_data->use_regular_non_max_suppression = false; in Init()
114 op_data->use_regular_non_max_suppression = m["use_regular_nms"].AsBool(); in Init()
116 op_data->non_max_suppression_score_threshold = in Init()
118 op_data->intersection_over_union_threshold = m["nms_iou_threshold"].AsFloat(); in Init()
119 op_data->num_classes = m["num_classes"].AsInt32(); in Init()
[all …]
Dwhile.cc117 auto* op_data = new OpData; in Init() local
119 op_data->cond_subgraph_index = params->cond_subgraph_index; in Init()
120 op_data->body_subgraph_index = params->body_subgraph_index; in Init()
121 op_data->cond_has_dynamic_output_tensors = false; in Init()
122 op_data->body_has_dynamic_output_tensors = false; in Init()
123 return op_data; in Init()
131 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
139 TF_LITE_ENSURE(context, op_data->cond_subgraph_index < subgraphs->size()); in Prepare()
140 TF_LITE_ENSURE(context, op_data->body_subgraph_index < subgraphs->size()); in Prepare()
142 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Prepare()
[all …]
Delementwise.cc97 auto* op_data = static_cast<OpData*>(node->user_data); in GenericPrepare() local
114 op_data->input_offset = input_params->zero_point->data[0]; in GenericPrepare()
115 op_data->output_offset = output_params->zero_point->data[0]; in GenericPrepare()
117 TF_LITE_ENSURE_EQ(context, op_data->input_offset, 0); in GenericPrepare()
118 TF_LITE_ENSURE_EQ(context, op_data->output_offset, 0); in GenericPrepare()
122 op_data->needs_rescale = input_scale != output_scale; in GenericPrepare()
123 if (op_name == kAbsName && op_data->needs_rescale) { in GenericPrepare()
124 SetAbsOutputMultiplier(input_scale, output_scale, &op_data->multiplier, in GenericPrepare()
125 &op_data->shift); in GenericPrepare()
127 SetRsqrtOutputMultiplier(input_scale, output_scale, &op_data->multiplier, in GenericPrepare()
[all …]
Dnumeric_verify.cc69 auto* op_data = new OpData(); in Init() local
70 op_data->float_input_initialized = false; in Init()
76 op_data->tolerance = tolerance; in Init()
77 op_data->log_if_failed = log_if_failed; in Init()
79 return op_data; in Init()
89 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
100 if (op_data->cache_tensor_id == kTensorNotAllocated) { in Prepare()
102 context, context->AddTensors(context, 1, &op_data->cache_tensor_id)); in Prepare()
107 node->temporaries->data[0] = op_data->cache_tensor_id; in Prepare()
143 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
[all …]
Dif.cc39 auto* op_data = new OpData; in Init() local
41 op_data->then_subgraph_index = params->then_subgraph_index; in Init()
42 op_data->else_subgraph_index = params->else_subgraph_index; in Init()
43 return op_data; in Init()
51 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
72 TF_LITE_ENSURE(context, op_data->then_subgraph_index < subgraphs->size()); in Prepare()
73 TF_LITE_ENSURE(context, op_data->else_subgraph_index < subgraphs->size()); in Prepare()
75 Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get(); in Prepare()
76 Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get(); in Prepare()
138 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
[all …]
Dsvdf.cc64 auto* op_data = new OpData(); in Init() local
65 op_data->float_weights_time_initialized = false; in Init()
68 &op_data->scratch_tensor_index); in Init()
69 return op_data; in Init()
78 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
79 int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare()
168 op_data->compute_row_sums = true; in Prepare()
274 QuantizeMultiplier(effective_scale_1, &op_data->effective_scale_1_a, in Prepare()
275 &op_data->effective_scale_1_b); in Prepare()
276 QuantizeMultiplier(effective_scale_2, &op_data->effective_scale_2_a, in Prepare()
[all …]
Ddequantize.cc43 auto* op_data = new OpData(); in Init() local
44 op_data->float_dequantized_weights_initialized = false; in Init()
45 return op_data; in Init()
79 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
82 op_data->float_dequantized_weights_initialized) { in Eval()
93 op_data->float_dequantized_weights_initialized = true; in Eval()
Dlstm.cc175 OpData* op_data = static_cast<OpData*>(node->user_data); in PopulateQuantizedLstmParams8x8_16() local
176 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16()
839 auto* op_data = new OpData(); in Init() local
840 op_data->kernel_type = kTfLiteLSTMFullKernel; in Init()
843 &op_data->scratch_tensor_index); in Init()
846 &op_data->ledger_index); in Init()
847 return op_data; in Init()
1153 OpData* op_data, in PopulatePrecomputedZPTimesWeightsWithBias() argument
1200 &op_data->integer_lstm_param; in PopulatePrecomputedZPTimesWeightsWithBias()
1212 const bool is_layer_norm = op_data->use_layer_norm; in PopulatePrecomputedZPTimesWeightsWithBias()
[all …]
Ddensify.cc44 auto* op_data = new OpData(); in Init() local
45 op_data->dense_weights_initialized = false; in Init()
46 return op_data; in Init()
71 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
73 if (op_data->dense_weights_initialized) { in Eval()
106 op_data->dense_weights_initialized = true; in Eval()
Dbatch_matmul.cc84 auto* op_data = new OpData(); in Init() local
86 op_data->rhs_transposed = false; in Init()
91 &op_data->scratch_tensor_index); in Init()
92 return op_data; in Init()
129 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in InitializeTemporaries() local
154 node->temporaries->data[0] = op_data->scratch_tensor_index; in InitializeTemporaries()
177 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in InitializeTemporaries()
217 op_data->compute_row_sums = true; in InitializeTemporaries()
218 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in InitializeTemporaries()
230 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in InitializeTemporaries()
[all …]
Dbasic_rnn.cc48 auto* op_data = new OpData(); in Init() local
50 &op_data->scratch_tensor_index); in Init()
51 return op_data; in Init()
112 auto* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
113 op_data->compute_row_sums = true; in Prepare()
116 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
127 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in Prepare()
141 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in Prepare()
154 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in Prepare()
169 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in Prepare()
[all …]
Dunidirectional_sequence_rnn.cc49 auto* op_data = new OpData(); in Init() local
51 &op_data->scratch_tensor_index); in Init()
52 return op_data; in Init()
119 auto* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
120 op_data->compute_row_sums = true; in Prepare()
123 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
134 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in Prepare()
148 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in Prepare()
161 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in Prepare()
176 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in Prepare()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsoftmax.cc32 const SoftmaxParams& op_data) { in SoftmaxQuantized() argument
35 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
42 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
56 depth, op_data.input_multiplier, op_data.input_left_shift, in SoftmaxQuantized()
57 op_data.diff_min, in SoftmaxQuantized()
62 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
/external/tensorflow/tensorflow/core/ops/
Dshape_function_test.cc26 std::vector<OpRegistrationData> op_data; in TEST() local
27 op_registry->GetOpRegistrationData(&op_data); in TEST()
28 for (const OpRegistrationData& op_reg_data : op_data) { in TEST()

12