Home
last modified time | relevance | path

Searched refs:GetTensorData (Results 1 – 25 of 161) sorted by relevance

1234567

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_eval.cc1795 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalFloat()
1824 const float* input_ptr = GetTensorData<float>(input) + t_rel * input_step; in EvalFloat()
1827 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalFloat()
1830 GetTensorData<float>(output) + t_rel * output_step + output_offset; in EvalFloat()
1833 input_ptr, GetTensorData<float>(input_to_input_weights), in EvalFloat()
1834 GetTensorData<float>(input_to_forget_weights), in EvalFloat()
1835 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
1836 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
1837 GetTensorData<float>(aux_input_to_input_weights), in EvalFloat()
1838 GetTensorData<float>(aux_input_to_forget_weights), in EvalFloat()
[all …]
H A Dfully_connected.cc491 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), num_units, in EvalPie()
493 GetTensorData<float>(output)); in EvalPie()
495 std::fill_n(GetTensorData<float>(output), batch_size * num_units, 0.0f); in EvalPie()
500 GetTensorData<float>(filter), num_units, input_size, in EvalPie()
501 GetTensorData<float>(input), batch_size, GetTensorData<float>(output)); in EvalPie()
505 GetTensorData<float>(output), batch_size * num_units, params->activation, in EvalPie()
506 GetTensorData<float>(output)); in EvalPie()
529 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), num_units, in EvalHybridDense()
531 GetTensorData<float>(output)); in EvalHybridDense()
533 std::fill_n(GetTensorData<float>(output), batch_size * num_units, 0.0f); in EvalHybridDense()
[all …]
H A Dactivations.cc147 uint8_t* output_data = GetTensorData<uint8_t>(output); in EvalUsingLookupTable()
148 const uint8_t* input_data = GetTensorData<uint8_t>(input); in EvalUsingLookupTable()
196 optimized_ops::ReluX(params, GetTensorShape(input), GetTensorData<T>(input), in QuantizedReluX()
197 GetTensorShape(output), GetTensorData<T>(output)); in QuantizedReluX()
775 optimized_ops::Relu(GetTensorShape(input), GetTensorData<float>(input), in ReluEval()
776 GetTensorShape(output), GetTensorData<float>(output)); in ReluEval()
810 optimized_ops::Relu1(GetTensorShape(input), GetTensorData<float>(input), in Relu1Eval()
812 GetTensorData<float>(output)); in Relu1Eval()
844 GetTensorShape(input), GetTensorData<float>(input), in HardSwishEval()
845 GetTensorShape(output), GetTensorData<float>(output)); in HardSwishEval()
[all …]
H A Dtranspose_conv.cc115 shape->data[i] = GetTensorData<int32_t>(shape_tensor)[i]; in ResizeTensor()
215 GetTensorData<float>(weights), in ResizeAndTransposeWeights()
217 GetTensorData<float>(transposed_weights)); in ResizeAndTransposeWeights()
220 GetTensorData<uint8>(weights), in ResizeAndTransposeWeights()
222 GetTensorData<uint8>(transposed_weights)); in ResizeAndTransposeWeights()
226 GetTensorData<int8>(weights), in ResizeAndTransposeWeights()
228 GetTensorData<int8>(transposed_weights)); in ResizeAndTransposeWeights()
413 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
414 GetTensorShape(weights), GetTensorData<float>(weights), in EvalFloat()
415 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat()
[all …]
H A Dbidirectional_sequence_rnn.cc396 const float* fw_bias_ptr = GetTensorData<float>(fw_bias); in EvalFloat()
397 const float* fw_input_weights_ptr = GetTensorData<float>(fw_input_weights); in EvalFloat()
399 GetTensorData<float>(fw_recurrent_weights); in EvalFloat()
402 const float* bw_bias_ptr = GetTensorData<float>(bw_bias); in EvalFloat()
403 const float* bw_input_weights_ptr = GetTensorData<float>(bw_input_weights); in EvalFloat()
405 GetTensorData<float>(bw_recurrent_weights); in EvalFloat()
409 ? GetTensorData<float>(fw_aux_input_weights) in EvalFloat()
413 ? GetTensorData<float>(bw_aux_input_weights) in EvalFloat()
422 float* fw_hidden_state_ptr_batch = GetTensorData<float>(fw_hidden_state); in EvalFloat()
425 GetTensorData<float>(input) + s * input_size * batch_size; in EvalFloat()
[all …]
H A Dconv.cc161 const float* input_data = GetTensorData<float>(input); in TransposeFloatTensor()
162 float* output_data = GetTensorData<float>(output); in TransposeFloatTensor()
705 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
706 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
707 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized()
708 GetTensorShape(output), GetTensorData<uint8_t>(output), in EvalQuantized()
709 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), in EvalQuantized()
718 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
719 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
720 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized()
[all …]
H A Dreduce.cc109 const int* axis = GetTensorData<int>(op_context->axis); in ResizeOutputTensor()
382 GetTensorData<T>(op_context->input), &op_context->input->dims->data[0], in Mean()
383 op_context->input->dims->size, GetTensorData<T>(op_context->output), in Mean()
385 GetTensorData<int>(op_context->axis), num_axis, in Mean()
402 GetTensorData<T>(op_context->input), op_context->input->params.zero_point, in QuantizedMeanOrSum()
404 op_context->input->dims->size, GetTensorData<T>(op_context->output), in QuantizedMeanOrSum()
407 GetTensorData<int>(op_context->axis), num_axis, in QuantizedMeanOrSum()
432 ResolveAxis(GetTensorData<int>(op_context.axis), num_axis, &op_params); in EvalIntegerMean()
438 GetTensorData<int>(temp_index), in EvalIntegerMean()
439 GetTensorData<int>(resolved_axis), in EvalIntegerMean()
[all …]
H A Ddequantize.h67 GetTensorData<uint8_t>(input), GetTensorShape(output), in PerChannelDequantizeImpl()
68 GetTensorData<float>(output)); in PerChannelDequantizeImpl()
73 GetTensorData<int8_t>(input), GetTensorShape(output), in PerChannelDequantizeImpl()
74 GetTensorData<float>(output)); in PerChannelDequantizeImpl()
97 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in DequantizeImpl()
98 GetTensorShape(output), GetTensorData<float>(output)); in DequantizeImpl()
101 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in DequantizeImpl()
102 GetTensorShape(output), GetTensorData<float>(output)); in DequantizeImpl()
108 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in DequantizeImpl()
109 GetTensorShape(output), GetTensorData<float>(output)); in DequantizeImpl()
[all …]
H A Dsvdf.cc319 params, GetTensorShape(input), GetTensorData<float>(input), in Eval()
321 GetTensorData<float>(weights_feature), GetTensorShape(weights_time), in Eval()
322 GetTensorData<float>(weights_time), GetTensorShape(bias), in Eval()
323 GetTensorData<float>(bias), GetTensorData<float>(scratch), in Eval()
324 GetTensorData<float>(state), GetTensorShape(output), in Eval()
325 GetTensorData<float>(output)); in Eval()
354 const int8_t* weights_time_ptr = GetTensorData<int8_t>(weights_time); in Eval()
356 GetTensorData<float>(float_weights_time); in Eval()
367 zero_points_ptr = GetTensorData<int32_t>(zero_points); in Eval()
368 row_sums_ptr = GetTensorData<int32_t>(row_sums); in Eval()
[all …]
H A Dquantize.cc170 const float* input_data = GetTensorData<float>(input); in Eval()
188 GetTensorData<int8_t>(output)); in Eval()
193 GetTensorData<uint8_t>(output)); in Eval()
198 GetTensorData<int16_t>(output)); in Eval()
214 GetTensorData<int8_t>(output)); in Eval()
219 GetTensorData<uint8_t>(output)); in Eval()
224 GetTensorData<int16_t>(output)); in Eval()
239 Requantize<kernel_type>(GetTensorData<int32_t>(input), in Eval()
244 GetTensorData<int8_t>(output)); in Eval()
247 Requantize<kernel_type>(GetTensorData<int32_t>(input), in Eval()
[all …]
H A Dvariable_ops_test.cc152 GetTensorData<float>(input_data_index)[0] = 1717; in TEST_F()
158 EXPECT_EQ(GetTensorData<float>(output)[0], 1717); in TEST_F()
165 GetTensorData<float>(input_data_index)[0] = 1.0; in TEST_F()
166 GetTensorData<float>(input_data_index)[1] = 2.0; in TEST_F()
167 GetTensorData<float>(input_data_index)[2] = 3.0; in TEST_F()
168 GetTensorData<float>(input_data_index)[3] = 4.0; in TEST_F()
174 EXPECT_EQ(GetTensorData<float>(output)[0], 1.0); in TEST_F()
175 EXPECT_EQ(GetTensorData<float>(output)[1], 2.0); in TEST_F()
176 EXPECT_EQ(GetTensorData<float>(output)[2], 3.0); in TEST_F()
177 EXPECT_EQ(GetTensorData<float>(output)[3], 4.0); in TEST_F()
[all …]
H A Ddepthwise_conv.cc321 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
322 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat()
323 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat()
324 GetTensorShape(output), GetTensorData<float>(output)); in EvalFloat()
327 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
328 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat()
329 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat()
330 GetTensorShape(output), GetTensorData<float>(output), in EvalFloat()
365 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
366 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
[all …]
H A Dresize_nearest_neighbor.cc52 const int32* size_data = GetTensorData<int32>(size); in ResizeOutputTensor()
111 op_params, GetTensorShape(input), GetTensorData<int32>(input), in Eval()
112 GetTensorShape(size), GetTensorData<int32>(size), in Eval()
113 GetTensorShape(output), GetTensorData<int32>(output)); in Eval()
117 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in Eval()
118 GetTensorShape(size), GetTensorData<int32>(size), in Eval()
119 GetTensorShape(output), GetTensorData<uint8_t>(output)); in Eval()
123 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in Eval()
124 GetTensorShape(size), GetTensorData<int32>(size), in Eval()
125 GetTensorShape(output), GetTensorData<uint8_t>(output)); in Eval()
[all …]
H A Dunidirectional_sequence_rnn.cc214 const float* bias_ptr = GetTensorData<float>(bias); in EvalFloat()
225 const float* input_weights_ptr = GetTensorData<float>(input_weights); in EvalFloat()
226 const float* recurrent_weights_ptr = GetTensorData<float>(recurrent_weights); in EvalFloat()
230 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalFloat()
235 GetTensorData<float>(input) + s * input_size * batch_size; in EvalFloat()
237 GetTensorData<float>(output) + s * num_units * batch_size; in EvalFloat()
249 GetTensorData<float>(hidden_state) + b * num_units; in EvalFloat()
252 const float* input_ptr_batch = GetTensorData<float>(input) + in EvalFloat()
255 float* output_ptr_batch = GetTensorData<float>(output) + in EvalFloat()
285 const float* bias_ptr = GetTensorData<float>(bias); in EvalHybrid()
[all …]
H A Dbatch_matmul.cc415 TransposeRowsColumnsImpl<float>(tensor_in, GetTensorData<float>(tensor_in), in TransposeRowsColumns()
417 GetTensorData<float>(tensor_out)); in TransposeRowsColumns()
421 tensor_in, GetTensorData<int8_t>(tensor_in), tensor_out, in TransposeRowsColumns()
422 GetTensorData<int8_t>(tensor_out)); in TransposeRowsColumns()
426 tensor_in, GetTensorData<int16_t>(tensor_in), tensor_out, in TransposeRowsColumns()
427 GetTensorData<int16_t>(tensor_out)); in TransposeRowsColumns()
470 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors); in EvalHybrid()
473 input_offset_ptr = GetTensorData<int32_t>(input_offsets); in EvalHybrid()
474 row_sums_ptr = GetTensorData<int32_t>(row_sums); in EvalHybrid()
478 int8_t* quant_data = GetTensorData<int8_t>(input_quantized); in EvalHybrid()
[all …]
H A Dwhere.cc43 const T* cond_data = GetTensorData<T>(cond_tensor); in ResizeOutputTensor()
163 GetTensorData<bool>(cond_tensor), in Eval()
164 GetTensorData<int64_t>(output)); in Eval()
168 GetTensorData<float>(cond_tensor), in Eval()
169 GetTensorData<int64_t>(output)); in Eval()
173 GetTensorData<int64_t>(cond_tensor), in Eval()
174 GetTensorData<int64_t>(output)); in Eval()
178 GetTensorData<int32_t>(cond_tensor), in Eval()
179 GetTensorData<int64_t>(output)); in Eval()
183 GetTensorData<int8_t>(cond_tensor), in Eval()
[all …]
H A Dreverse.cc80 int axis = GetTensorData<int32_t>(axis_tensor)[0]; in Eval()
94 axis, GetTensorShape(input), GetTensorData<float>(input), in Eval()
95 GetTensorShape(output), GetTensorData<float>(output)); in Eval()
101 axis, GetTensorShape(input), GetTensorData<uint8_t>(input), in Eval()
102 GetTensorShape(output), GetTensorData<uint8_t>(output)); in Eval()
107 axis, GetTensorShape(input), GetTensorData<int16_t>(input), in Eval()
108 GetTensorShape(output), GetTensorData<int16_t>(output)); in Eval()
113 axis, GetTensorShape(input), GetTensorData<int32_t>(input), in Eval()
114 GetTensorShape(output), GetTensorData<int32_t>(output)); in Eval()
119 axis, GetTensorShape(input), GetTensorData<int64_t>(input), in Eval()
[all …]
H A Dmaximum_minimum.cc98 GetTensorData<data_type>(op_context.input1), in TFLiteOperation()
100 GetTensorData<data_type>(op_context.input2), in TFLiteOperation()
102 GetTensorData<data_type>(op_context.output), in TFLiteOperation()
117 GetTensorData<int8>(op_context.input1), in TFLiteOperation()
119 GetTensorData<int8>(op_context.input2), in TFLiteOperation()
121 GetTensorData<int8>(op_context.output), MaximumOp::template op<int8>); in TFLiteOperation()
125 GetTensorShape(op_context.input1), GetTensorData<int8>(op_context.input1), in TFLiteOperation()
126 GetTensorShape(op_context.input2), GetTensorData<int8>(op_context.input2), in TFLiteOperation()
127 GetTensorShape(op_context.output), GetTensorData<int8>(op_context.output), in TFLiteOperation()
142 GetTensorData<int8>(op_context.input1), in TFLiteOperation()
[all …]
H A Dbasic_rnn.cc213 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalFloat()
215 const float* input_ptr_batch = GetTensorData<float>(input); in EvalFloat()
216 float* output_ptr_batch = GetTensorData<float>(output); in EvalFloat()
218 const float* input_weights_ptr = GetTensorData<float>(input_weights); in EvalFloat()
219 const float* recurrent_weights_ptr = GetTensorData<float>(recurrent_weights); in EvalFloat()
220 const float* bias_ptr = GetTensorData<float>(bias); in EvalFloat()
246 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalHybrid()
248 const float* input_ptr_batch = GetTensorData<float>(input); in EvalHybrid()
249 float* output_ptr_batch = GetTensorData<float>(output); in EvalHybrid()
251 const int8_t* input_weights_ptr = GetTensorData<int8_t>(input_weights); in EvalHybrid()
[all …]
H A Dpad.cc55 const int32* paddings_data = GetTensorData<int32>(paddings); in PadContext()
83 const int32* paddings_data = GetTensorData<int32>(op_context->paddings); in ResizeOutputTensor()
96 paddings_data = GetTensorData<int32>(op_context->paddings); in ResizeOutputTensor()
155 pad_value = *GetTensorData<integer_type>(op_context.constant_values); in EvalInt()
161 GetTensorData<integer_type>(op_context.input), &pad_value_copy, in EvalInt()
163 GetTensorData<integer_type>(op_context.output)); in EvalInt()
166 GetTensorData<integer_type>(op_context.input), in EvalInt()
168 GetTensorData<integer_type>(op_context.output)); in EvalInt()
189 const int32* paddings_data = GetTensorData<int32>(op_context.paddings); in Eval()
207 GetTensorData<scalar>(op_context.input), &pad_value_copy, \ in Eval()
[all …]
H A Dsquared_difference.cc183 GetTensorData<T>(input1), GetTensorShape(input2), in EvalQuantizedSquaredDifference()
184 GetTensorData<T>(input2), GetTensorShape(output), in EvalQuantizedSquaredDifference()
185 GetTensorData<T>(output), reference_integer_ops::CheckArithmeticParams, in EvalQuantizedSquaredDifference()
190 flat_size, op_data->arithmetic_params, GetTensorData<int8_t>(input1), in EvalQuantizedSquaredDifference()
191 GetTensorData<int8_t>(input2), GetTensorData<int8_t>(output), in EvalQuantizedSquaredDifference()
202 GetTensorShape(input1), GetTensorData<T>(input1), in EvalSquaredDifference()
203 GetTensorShape(input2), GetTensorData<T>(input2), in EvalSquaredDifference()
204 GetTensorShape(output), GetTensorData<T>(output), SquaredDifference<T>); in EvalSquaredDifference()
207 GetTensorShape(input1), GetTensorData<T>(input1), in EvalSquaredDifference()
208 GetTensorShape(input2), GetTensorData<T>(input2), in EvalSquaredDifference()
[all …]
H A Dpooling.cc137 GetTensorData<float>(input), \ in AverageEvalFloat()
139 GetTensorData<float>(output))) in AverageEvalFloat()
169 GetTensorData<uint8_t>(input), \ in AverageEvalQuantizedUint8()
171 GetTensorData<uint8_t>(output))) in AverageEvalQuantizedUint8()
202 GetTensorData<int8_t>(input), \ in AverageEvalQuantizedInt8()
204 GetTensorData<int8_t>(output))) in AverageEvalQuantizedInt8()
234 GetTensorData<int16_t>(input), \ in AverageEvalQuantizedInt16()
236 GetTensorData<int16_t>(output))) in AverageEvalQuantizedInt16()
259 type::MaxPool(op_params, GetTensorShape(input), GetTensorData<float>(input), \ in MaxEvalFloat()
260 GetTensorShape(output), GetTensorData<float>(output)) in MaxEvalFloat()
[all …]
H A Dsub.cc315 op_params, GetTensorShape(input1), GetTensorData<data_type>(input1), in EvalSubImpl()
316 GetTensorShape(input2), GetTensorData<data_type>(input2), in EvalSubImpl()
317 GetTensorShape(output), GetTensorData<data_type>(output)); in EvalSubImpl()
320 op_params, GetTensorShape(input1), GetTensorData<data_type>(input1), in EvalSubImpl()
321 GetTensorShape(input2), GetTensorData<data_type>(input2), in EvalSubImpl()
322 GetTensorShape(output), GetTensorData<data_type>(output)); in EvalSubImpl()
329 op_params, GetTensorShape(input1), GetTensorData<data_type>(input1), in EvalSubImpl()
330 GetTensorShape(input2), GetTensorData<data_type>(input2), in EvalSubImpl()
331 GetTensorShape(output), GetTensorData<data_type>(output)); in EvalSubImpl()
334 op_params, GetTensorShape(input1), GetTensorData<data_type>(input1), in EvalSubImpl()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc392 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalFloat()
418 const float* input_ptr = GetTensorData<float>(input) + t_rel * input_step; in EvalFloat()
421 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalFloat()
424 GetTensorData<float>(output) + t_rel * output_step + output_offset; in EvalFloat()
427 input_ptr, GetTensorData<float>(input_to_input_weights), in EvalFloat()
428 GetTensorData<float>(input_to_forget_weights), in EvalFloat()
429 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
430 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat()
431 GetTensorData<float>(aux_input_to_input_weights), in EvalFloat()
432 GetTensorData<float>(aux_input_to_forget_weights), in EvalFloat()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc315 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalCalibration()
341 const float* input_ptr = GetTensorData<float>(input) + t_rel * input_step; in EvalCalibration()
344 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalCalibration()
347 GetTensorData<float>(output) + t_rel * output_step + output_offset; in EvalCalibration()
350 input_ptr, GetTensorData<float>(input_to_input_weights), in EvalCalibration()
351 GetTensorData<float>(input_to_forget_weights), in EvalCalibration()
352 GetTensorData<float>(input_to_cell_weights), in EvalCalibration()
353 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalCalibration()
354 GetTensorData<float>(aux_input_to_input_weights), in EvalCalibration()
355 GetTensorData<float>(aux_input_to_forget_weights), in EvalCalibration()
[all …]

1234567