Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc1736 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 …]
Dfully_connected.cc415 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 …]
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 …]
Dreduce.cc106 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 …]
Dactivations.cc145 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 …]
Dbidirectional_sequence_rnn.cc394 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 …]
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 …]
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()
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 …]
Dsvdf.cc317 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 …]
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 …]
Dconv.cc158 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 …]
Ddepthwise_conv.cc319 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 …]
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 …]
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 …]
Dquantize.cc166 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 …]
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 …]
Dbatch_matmul.cc412 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 …]
Dvariable_ops_test.cc120 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()
Dbasic_rnn.cc212 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 …]
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 …]
Dcumsum.cc63 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()
Dneg.cc57 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()
Dsquared_difference.cc181 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/
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 …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
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 …]

123456