/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | lstm_eval.cc | 1795 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 D | fully_connected.cc | 491 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 D | activations.cc | 147 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 D | transpose_conv.cc | 115 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 D | bidirectional_sequence_rnn.cc | 396 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 D | conv.cc | 161 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 D | reduce.cc | 109 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 D | dequantize.h | 67 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 D | svdf.cc | 319 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 D | quantize.cc | 170 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 D | variable_ops_test.cc | 152 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 D | depthwise_conv.cc | 321 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 D | resize_nearest_neighbor.cc | 52 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 D | unidirectional_sequence_rnn.cc | 214 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 D | batch_matmul.cc | 415 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 D | where.cc | 43 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 D | reverse.cc | 80 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 D | maximum_minimum.cc | 98 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 D | basic_rnn.cc | 213 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 D | pad.cc | 55 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 D | squared_difference.cc | 183 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 D | pooling.cc | 137 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 D | sub.cc | 315 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 D | lstm.cc | 392 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 D | lstm.cc | 315 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 …]
|