/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_eval.cc | 1736 float* scratch_buffer_ptr = GetTensorData<float>(scratch_buffer); in EvalFloat() 1762 const float* input_ptr = GetTensorData<float>(input) + t_rel * input_step; in EvalFloat() 1765 aux_input_ptr = GetTensorData<float>(aux_input) + t_rel * input_step; in EvalFloat() 1768 GetTensorData<float>(output) + t_rel * output_step + output_offset; in EvalFloat() 1771 input_ptr, GetTensorData<float>(input_to_input_weights), in EvalFloat() 1772 GetTensorData<float>(input_to_forget_weights), in EvalFloat() 1773 GetTensorData<float>(input_to_cell_weights), in EvalFloat() 1774 GetTensorData<float>(input_to_output_weights), aux_input_ptr, in EvalFloat() 1775 GetTensorData<float>(aux_input_to_input_weights), in EvalFloat() 1776 GetTensorData<float>(aux_input_to_forget_weights), in EvalFloat() [all …]
|
D | fully_connected.cc | 415 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), num_units, in EvalPie() 417 GetTensorData<float>(output)); in EvalPie() 419 std::fill_n(GetTensorData<float>(output), batch_size * num_units, 0.0f); in EvalPie() 424 GetTensorData<float>(filter), num_units, input_size, in EvalPie() 425 GetTensorData<float>(input), batch_size, GetTensorData<float>(output)); in EvalPie() 429 GetTensorData<float>(output), batch_size * num_units, params->activation, in EvalPie() 430 GetTensorData<float>(output)); in EvalPie() 453 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), num_units, in EvalHybridDense() 455 GetTensorData<float>(output)); in EvalHybridDense() 457 std::fill_n(GetTensorData<float>(output), batch_size * num_units, 0.0f); in EvalHybridDense() [all …]
|
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 …]
|
D | reduce.cc | 106 const int* axis = GetTensorData<int>(op_context->axis); in ResizeOutputTensor() 356 ResolveAxis(GetTensorData<int>(op_context.axis), num_axis, &op_params); in EvalMeanReferenceOps() 367 GetTensorData<uint8_t>(op_context.input), in EvalMeanReferenceOps() 371 GetTensorData<uint8_t>(op_context.output), in EvalMeanReferenceOps() 377 GetTensorData<integer_type>(input), in EvalMeanReferenceOps() 380 GetTensorData<integer_type>(op_context.output), in EvalMeanReferenceOps() 388 GetTensorData<integer_type>(input), input->dims->data, in EvalMeanReferenceOps() 389 input->dims->size, GetTensorData<integer_type>(op_context.output), in EvalMeanReferenceOps() 391 GetTensorData<int>(op_context.axis), num_axis, in EvalMeanReferenceOps() 392 op_context.params->keep_dims, GetTensorData<int>(temp_index), in EvalMeanReferenceOps() [all …]
|
D | activations.cc | 145 uint8_t* output_data = GetTensorData<uint8_t>(output); in EvalUsingLookupTable() 146 const uint8_t* input_data = GetTensorData<uint8_t>(input); in EvalUsingLookupTable() 194 optimized_ops::ReluX(params, GetTensorShape(input), GetTensorData<T>(input), in QuantizedReluX() 195 GetTensorShape(output), GetTensorData<T>(output)); in QuantizedReluX() 773 optimized_ops::Relu(GetTensorShape(input), GetTensorData<float>(input), in ReluEval() 774 GetTensorShape(output), GetTensorData<float>(output)); in ReluEval() 808 optimized_ops::Relu1(GetTensorShape(input), GetTensorData<float>(input), in Relu1Eval() 810 GetTensorData<float>(output)); in Relu1Eval() 842 GetTensorShape(input), GetTensorData<float>(input), in HardSwishEval() 843 GetTensorShape(output), GetTensorData<float>(output)); in HardSwishEval() [all …]
|
D | bidirectional_sequence_rnn.cc | 394 const float* fw_bias_ptr = GetTensorData<float>(fw_bias); in EvalFloat() 395 const float* fw_input_weights_ptr = GetTensorData<float>(fw_input_weights); in EvalFloat() 397 GetTensorData<float>(fw_recurrent_weights); in EvalFloat() 400 const float* bw_bias_ptr = GetTensorData<float>(bw_bias); in EvalFloat() 401 const float* bw_input_weights_ptr = GetTensorData<float>(bw_input_weights); in EvalFloat() 403 GetTensorData<float>(bw_recurrent_weights); in EvalFloat() 407 ? GetTensorData<float>(fw_aux_input_weights) in EvalFloat() 411 ? GetTensorData<float>(bw_aux_input_weights) in EvalFloat() 420 float* fw_hidden_state_ptr_batch = GetTensorData<float>(fw_hidden_state); in EvalFloat() 423 GetTensorData<float>(input) + s * input_size * batch_size; in EvalFloat() [all …]
|
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 …]
|
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() 412 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat() 413 GetTensorShape(weights), GetTensorData<float>(weights), in EvalFloat() 414 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat() [all …]
|
D | svdf.cc | 317 params, GetTensorShape(input), GetTensorData<float>(input), in Eval() 319 GetTensorData<float>(weights_feature), GetTensorShape(weights_time), in Eval() 320 GetTensorData<float>(weights_time), GetTensorShape(bias), in Eval() 321 GetTensorData<float>(bias), GetTensorData<float>(scratch), in Eval() 322 GetTensorData<float>(state), GetTensorShape(output), in Eval() 323 GetTensorData<float>(output)); in Eval() 352 const int8_t* weights_time_ptr = GetTensorData<int8_t>(weights_time); in Eval() 354 GetTensorData<float>(float_weights_time); in Eval() 365 zero_points_ptr = GetTensorData<int32_t>(zero_points); in Eval() 366 row_sums_ptr = GetTensorData<int32_t>(row_sums); in Eval() [all …]
|
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 …]
|
D | conv.cc | 158 const float* input_data = GetTensorData<float>(input); in TransposeFloatTensor() 159 float* output_data = GetTensorData<float>(output); in TransposeFloatTensor() 676 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized() 677 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized() 678 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized() 679 GetTensorShape(output), GetTensorData<uint8_t>(output), in EvalQuantized() 680 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), in EvalQuantized() 689 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized() 690 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized() 691 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized() [all …]
|
D | depthwise_conv.cc | 319 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat() 320 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat() 321 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat() 322 GetTensorShape(output), GetTensorData<float>(output)); in EvalFloat() 325 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat() 326 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat() 327 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat() 328 GetTensorShape(output), GetTensorData<float>(output), in EvalFloat() 363 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized() 364 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized() [all …]
|
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 …]
|
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 …]
|
D | quantize.cc | 166 const float* input_data = GetTensorData<float>(input); in Eval() 171 GetTensorData<int8_t>(output)); in Eval() 176 GetTensorData<uint8_t>(output)); in Eval() 181 GetTensorData<int16_t>(output)); in Eval() 192 Requantize<kernel_type>(GetTensorData<int16_t>(input), in Eval() 197 GetTensorData<int8_t>(output)); in Eval() 200 Requantize<kernel_type>(GetTensorData<int16_t>(input), in Eval() 205 GetTensorData<int16_t>(output)); in Eval() 211 Requantize<kernel_type>(GetTensorData<int16_t>(input), in Eval() 216 GetTensorData<int32_t>(output)); in Eval() [all …]
|
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 …]
|
D | batch_matmul.cc | 412 TransposeRowsColumnsImpl<float>(tensor_in, GetTensorData<float>(tensor_in), in TransposeRowsColumns() 414 GetTensorData<float>(tensor_out)); in TransposeRowsColumns() 418 tensor_in, GetTensorData<int8_t>(tensor_in), tensor_out, in TransposeRowsColumns() 419 GetTensorData<int8_t>(tensor_out)); in TransposeRowsColumns() 423 tensor_in, GetTensorData<int16_t>(tensor_in), tensor_out, in TransposeRowsColumns() 424 GetTensorData<int16_t>(tensor_out)); in TransposeRowsColumns() 467 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors); in EvalHybrid() 470 input_offset_ptr = GetTensorData<int32_t>(input_offsets); in EvalHybrid() 471 row_sums_ptr = GetTensorData<int32_t>(row_sums); in EvalHybrid() 475 int8_t* quant_data = GetTensorData<int8_t>(input_quantized); in EvalHybrid() [all …]
|
D | variable_ops_test.cc | 120 GetTensorData<float>(input_data_index)[0] = 1717; in TEST_F() 126 EXPECT_EQ(GetTensorData<float>(output)[0], 1717); in TEST_F() 133 GetTensorData<float>(input_data_index)[0] = 1717; in TEST_F() 145 GetTensorData<float>(input_data_index)[0] = 1717; in TEST_F() 151 EXPECT_EQ(GetTensorData<float>(output)[0], 1717); in TEST_F() 160 GetTensorData<float>(input_data_index)[0] = 1717; in TEST_F() 161 GetTensorData<float>(input_data_index)[1] = 2121; in TEST_F() 168 EXPECT_EQ(GetTensorData<float>(output)[0], 1717); in TEST_F() 169 EXPECT_EQ(GetTensorData<float>(output)[1], 2121); in TEST_F()
|
D | basic_rnn.cc | 212 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalFloat() 214 const float* input_ptr_batch = GetTensorData<float>(input); in EvalFloat() 215 float* output_ptr_batch = GetTensorData<float>(output); in EvalFloat() 217 const float* input_weights_ptr = GetTensorData<float>(input_weights); in EvalFloat() 218 const float* recurrent_weights_ptr = GetTensorData<float>(recurrent_weights); in EvalFloat() 219 const float* bias_ptr = GetTensorData<float>(bias); in EvalFloat() 245 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalHybrid() 247 const float* input_ptr_batch = GetTensorData<float>(input); in EvalHybrid() 248 float* output_ptr_batch = GetTensorData<float>(output); in EvalHybrid() 250 const int8_t* input_weights_ptr = GetTensorData<int8_t>(input_weights); in EvalHybrid() [all …]
|
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 …]
|
D | cumsum.cc | 63 int axis = *GetTensorData<int>(axis_tensor); in Eval() 73 optimized_ops::CumSum(GetTensorData<int>(input), GetTensorShape(input), in Eval() 75 GetTensorData<int>(output)); in Eval() 79 optimized_ops::CumSum(GetTensorData<int64_t>(input), in Eval() 81 params->reverse, GetTensorData<int64_t>(output)); in Eval() 85 optimized_ops::CumSum(GetTensorData<float>(input), GetTensorShape(input), in Eval() 87 GetTensorData<float>(output)); in Eval()
|
D | neg.cc | 57 GetTensorShape(input), GetTensorData<int64_t>(input), in Eval() 58 GetTensorShape(output), GetTensorData<int64_t>(output)); in Eval() 62 GetTensorShape(input), GetTensorData<int32_t>(input), in Eval() 63 GetTensorShape(output), GetTensorData<int32_t>(output)); in Eval() 66 reference_ops::Negate(GetTensorShape(input), GetTensorData<float>(input), in Eval() 68 GetTensorData<float>(output)); in Eval()
|
D | squared_difference.cc | 181 GetTensorData<T>(input1), GetTensorShape(input2), in EvalQuantizedSquaredDifference() 182 GetTensorData<T>(input2), GetTensorShape(output), in EvalQuantizedSquaredDifference() 183 GetTensorData<T>(output), reference_integer_ops::CheckArithmeticParams, in EvalQuantizedSquaredDifference() 188 flat_size, op_data->arithmetic_params, GetTensorData<int8_t>(input1), in EvalQuantizedSquaredDifference() 189 GetTensorData<int8_t>(input2), GetTensorData<int8_t>(output), in EvalQuantizedSquaredDifference() 200 GetTensorShape(input1), GetTensorData<T>(input1), in EvalSquaredDifference() 201 GetTensorShape(input2), GetTensorData<T>(input2), in EvalSquaredDifference() 202 GetTensorShape(output), GetTensorData<T>(output), SquaredDifference<T>); in EvalSquaredDifference() 205 GetTensorShape(input1), GetTensorData<T>(input1), in EvalSquaredDifference() 206 GetTensorShape(input2), GetTensorData<T>(input2), in EvalSquaredDifference() [all …]
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
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 …]
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
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 …]
|