Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_ctypes.h24 inline T* GetTensorData(TfLiteTensor* tensor);
27 inline float* GetTensorData(TfLiteTensor* tensor) { in GetTensorData() function
32 inline uint8_t* GetTensorData(TfLiteTensor* tensor) { in GetTensorData() function
37 inline int16_t* GetTensorData(TfLiteTensor* tensor) { in GetTensorData() function
42 inline int32_t* GetTensorData(TfLiteTensor* tensor) { in GetTensorData() function
47 inline int64_t* GetTensorData(TfLiteTensor* tensor) { in GetTensorData() function
52 inline bool* GetTensorData(TfLiteTensor* tensor) { in GetTensorData() function
57 inline int8_t* GetTensorData(TfLiteTensor* tensor) { in GetTensorData() function
62 inline const T* GetTensorData(const TfLiteTensor* tensor);
65 inline const float* GetTensorData(const TfLiteTensor* tensor) { in GetTensorData() function
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc384 optimized_ops::ReluX(params, GetTensorShape(input), GetTensorData<T>(input), in QuantizedRelu6()
385 GetTensorShape(output), GetTensorData<T>(output)); in QuantizedRelu6()
424 optimized_ops::Tanh(GetTensorShape(input), GetTensorData<float>(input), in TanhEval()
426 GetTensorData<float>(output)); in TanhEval()
428 reference_ops::Tanh(GetTensorShape(input), GetTensorData<float>(input), in TanhEval()
430 GetTensorData<float>(output)); in TanhEval()
439 params, GetTensorShape(input), GetTensorData<int16_t>(input), in TanhEval()
440 GetTensorShape(output), GetTensorData<int16_t>(output)); in TanhEval()
443 params, GetTensorShape(input), GetTensorData<int16_t>(input), in TanhEval()
444 GetTensorShape(output), GetTensorData<int16_t>(output)); in TanhEval()
[all …]
Dresize_nearest_neighbor.cc45 const int32* size_data = GetTensorData<int32>(size); in ResizeOutputTensor()
93 op_params, GetTensorShape(input), GetTensorData<int32>(input), in Eval()
94 GetTensorShape(size), GetTensorData<int32>(size), in Eval()
95 GetTensorShape(output), GetTensorData<int32>(output)); in Eval()
99 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in Eval()
100 GetTensorShape(size), GetTensorData<int32>(size), in Eval()
101 GetTensorShape(output), GetTensorData<uint8_t>(output)); in Eval()
105 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in Eval()
106 GetTensorShape(size), GetTensorData<int32>(size), in Eval()
107 GetTensorShape(output), GetTensorData<uint8_t>(output)); in Eval()
[all …]
Dreduce.cc100 const int* axis = GetTensorData<int>(op_context->axis); in ResizeOutputTensor()
291 ResolveAxis(GetTensorData<int>(op_context.axis), num_axis, &op_params); in EvalMean()
301 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalMean()
304 GetTensorData<uint8_t>(op_context.output), in EvalMean()
309 GetTensorData<float>(input), in EvalMean()
311 GetTensorData<float>(op_context.output)); in EvalMean()
320 ResolveAxis(GetTensorData<int>(op_context.axis), num_axis, &op_params); in EvalMean()
324 GetTensorData<int8_t>(input), op_context.input->params.zero_point, in EvalMean()
326 GetTensorData<int8_t>(op_context.output), in EvalMean()
333 GetTensorData<data_type>(op_context.input), \ in EvalMean()
[all …]
Dreverse.cc70 int axis = GetTensorData<int32_t>(axis_tensor)[0]; in Eval()
78 axis, GetTensorShape(input), GetTensorData<float>(input), in Eval()
79 GetTensorShape(output), GetTensorData<float>(output)); in Eval()
84 axis, GetTensorShape(input), GetTensorData<uint8_t>(input), in Eval()
85 GetTensorShape(output), GetTensorData<uint8_t>(output)); in Eval()
90 axis, GetTensorShape(input), GetTensorData<int16_t>(input), in Eval()
91 GetTensorShape(output), GetTensorData<int16_t>(output)); in Eval()
96 axis, GetTensorShape(input), GetTensorData<int32_t>(input), in Eval()
97 GetTensorShape(output), GetTensorData<int32_t>(output)); in Eval()
102 axis, GetTensorShape(input), GetTensorData<int64_t>(input), in Eval()
[all …]
Dconv.cc128 const float* input_data = GetTensorData<float>(input); in TransposeFloatTensor()
129 float* output_data = GetTensorData<float>(output); in TransposeFloatTensor()
461 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
462 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
463 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized()
464 GetTensorShape(output), GetTensorData<uint8_t>(output), in EvalQuantized()
465 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), gemm_context); in EvalQuantized()
488 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
489 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
490 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized()
[all …]
Drange.cc55 GetSize(context, *GetTensorData<int32_t>(start), in ResizeOutput()
56 *GetTensorData<int32_t>(limit), in ResizeOutput()
57 *GetTensorData<int32_t>(delta), &size)); in ResizeOutput()
61 TF_LITE_ENSURE_OK(context, GetSize(context, *GetTensorData<float>(start), in ResizeOutput()
62 *GetTensorData<float>(limit), in ResizeOutput()
63 *GetTensorData<float>(delta), &size)); in ResizeOutput()
121 const T start_value = *GetTensorData<T>(start); in EvalImpl()
122 const T delta_value = *GetTensorData<T>(delta); in EvalImpl()
123 T* output_data = GetTensorData<T>(output); in EvalImpl()
Dmul.cc120 GetTensorData<data_type>(input1), GetTensorShape(input2), \ in EvalMul()
121 GetTensorData<data_type>(input2), GetTensorShape(output), \ in EvalMul()
122 GetTensorData<data_type>(output)) in EvalMul()
175 GetTensorData<dtype>(input1), GetTensorShape(input2), \ in EvalQuantized()
176 GetTensorData<dtype>(input2), GetTensorShape(output), \ in EvalQuantized()
177 GetTensorData<dtype>(output)) in EvalQuantized()
206 GetTensorData<int16_t>(input1), GetTensorShape(input2), \ in EvalQuantized()
207 GetTensorData<int16_t>(input2), GetTensorShape(output), \ in EvalQuantized()
208 GetTensorData<int16_t>(output)) in EvalQuantized()
223 GetTensorData<int16_t>(input1), GetTensorShape(input2), \ in EvalQuantized()
[all …]
Dpow.cc84 GetTensorShape(input1), GetTensorData<T>(input1), in PowImpl()
85 GetTensorShape(input2), GetTensorData<T>(input2), in PowImpl()
86 GetTensorShape(output), GetTensorData<T>(output)); in PowImpl()
88 reference_ops::Pow(GetTensorShape(input1), GetTensorData<T>(input1), in PowImpl()
89 GetTensorShape(input2), GetTensorData<T>(input2), in PowImpl()
90 GetTensorShape(output), GetTensorData<T>(output)); in PowImpl()
96 const int32_t* data = GetTensorData<int32_t>(input); in CheckValue()
Dfully_connected.cc305 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in FullyConnectedInt8()
306 GetTensorShape(filter), GetTensorData<int8_t>(filter), in FullyConnectedInt8()
307 GetTensorShape(bias), GetTensorData<int32_t>(bias), in FullyConnectedInt8()
308 GetTensorShape(output), GetTensorData<int8_t>(output), gemm_context); in FullyConnectedInt8()
334 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), \ in EvalQuantized()
335 GetTensorShape(filter), GetTensorData<uint8_t>(filter), \ in EvalQuantized()
336 GetTensorShape(bias), GetTensorData<int32_t>(bias), \ in EvalQuantized()
337 GetTensorShape(output), GetTensorData<output_data_type>(output), \ in EvalQuantized()
413 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), \ in EvalShuffledQuantized()
414 GetTensorShape(filter), GetTensorData<uint8_t>(filter), \ in EvalShuffledQuantized()
[all …]
Dstrided_slice.cc83 const bool pos_stride = GetTensorData<int32_t>(op_context->strides)[idx] > 0; in GetBeginValueAtIndex()
86 : ClampedIndex(GetTensorData<int32_t>(op_context->begin)[idx], dim, in GetBeginValueAtIndex()
92 const bool pos_stride = GetTensorData<int32_t>(op_context->strides)[idx] > 0; in GetEndValueAtIndex()
95 : ClampedIndex(GetTensorData<int32_t>(op_context->end)[idx], dim, in GetEndValueAtIndex()
107 int32_t stride = GetTensorData<int32_t>(op_context->strides)[idx]; in ResizeOutputTensor()
197 starts.emplace_back(GetTensorData<int32_t>(op_context.begin)[idx]); in Eval()
198 stops.emplace_back(GetTensorData<int32_t>(op_context.end)[idx]); in Eval()
199 strides.emplace_back(GetTensorData<int32_t>(op_context.strides)[idx]); in Eval()
212 GetTensorData<data_type>(op_context.input), \ in Eval()
214 GetTensorData<data_type>(op_context.output)) in Eval()
Dslice.cc51 T size_value = GetTensorData<T>(size)[idx]; in CalculateOutputShapeVector()
57 size_value = SizeOfDimension(input, idx) - GetTensorData<T>(begin)[idx]; in CalculateOutputShapeVector()
60 GetTensorData<T>(begin)[idx] + size_value) { in CalculateOutputShapeVector()
75 begins->push_back(GetTensorData<T>(begin)[idx]); in GetBeginAndSizeVectors()
76 sizes->push_back(GetTensorData<T>(size)[idx]); in GetBeginAndSizeVectors()
189 op_params, GetTensorShape(input), GetTensorData<data_type>(input), \ in Eval()
190 GetTensorShape(output), GetTensorData<data_type>(output)); \ in Eval()
193 op_params, GetTensorShape(input), GetTensorData<data_type>(input), \ in Eval()
194 GetTensorShape(output), GetTensorData<data_type>(output)); \ in Eval()
Dlogical.cc90 GetTensorShape(input1), GetTensorData<bool>(input1), in LogicalImpl()
91 GetTensorShape(input2), GetTensorData<bool>(input2), in LogicalImpl()
92 GetTensorShape(output), GetTensorData<bool>(output), func); in LogicalImpl()
94 reference_ops::Logical(GetTensorShape(input1), GetTensorData<bool>(input1), in LogicalImpl()
95 GetTensorShape(input2), GetTensorData<bool>(input2), in LogicalImpl()
96 GetTensorShape(output), GetTensorData<bool>(output), in LogicalImpl()
Dtranspose_conv.cc85 shape->data[i] = GetTensorData<int32_t>(shape_tensor)[i]; in ResizeOutputTensor()
222 op_params, GetTensorShape(input), GetTensorData<float>(input), in Eval()
223 GetTensorShape(weights), GetTensorData<float>(weights), in Eval()
224 GetTensorShape(output), GetTensorData<float>(output), in Eval()
225 GetTensorShape(im2col), GetTensorData<float>(im2col)); in Eval()
230 op_params, GetTensorShape(input), GetTensorData<float>(input), in Eval()
231 GetTensorShape(weights), GetTensorData<float>(weights), in Eval()
232 GetTensorShape(output), GetTensorData<float>(output), in Eval()
233 GetTensorShape(im2col), GetTensorData<float>(im2col)); in Eval()
Ddepthwise_conv.cc218 depthwise_conv(op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
219 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat()
220 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat()
221 GetTensorShape(output), GetTensorData<float>(output)); in EvalFloat()
261 GetTensorData<uint8_t>(input), GetTensorShape(filter), in EvalQuantized()
262 GetTensorData<uint8_t>(filter), GetTensorShape(bias), in EvalQuantized()
263 GetTensorData<int32_t>(bias), GetTensorShape(output), in EvalQuantized()
264 GetTensorData<uint8_t>(output)); in EvalQuantized()
288 GetTensorData<int8>(input), GetTensorShape(filter), in EvalQuantizedPerChannel()
289 GetTensorData<int8>(filter), GetTensorShape(bias), in EvalQuantizedPerChannel()
[all …]
Dcomparisons.cc92 GetTensorData<input_dtype>(input1), GetTensorShape(input2), \
93 GetTensorData<input_dtype>(input2), GetTensorShape(output), \
94 GetTensorData<bool>(output)); \
98 GetTensorData<input_dtype>(input1), GetTensorShape(input2), \
99 GetTensorData<input_dtype>(input2), GetTensorShape(output), \
100 GetTensorData<bool>(output)); \
117 op_params, GetTensorShape(input1), GetTensorData<type>(input1), \
118 GetTensorShape(input2), GetTensorData<type>(input2), \
119 GetTensorShape(output), GetTensorData<bool>(output)) \
121 op_params, GetTensorShape(input1), GetTensorData<type>(input1), \
[all …]
Dsquared_difference.cc85 GetTensorShape(input1), GetTensorData<T>(input1), in EvalSquaredDifference()
86 GetTensorShape(input2), GetTensorData<T>(input2), in EvalSquaredDifference()
87 GetTensorShape(output), GetTensorData<T>(output), SquaredDifference<T>); in EvalSquaredDifference()
90 GetTensorShape(input1), GetTensorData<T>(input1), in EvalSquaredDifference()
91 GetTensorShape(input2), GetTensorData<T>(input2), in EvalSquaredDifference()
92 GetTensorShape(output), GetTensorData<T>(output), SquaredDifference<T>); in EvalSquaredDifference()
Dpad.cc50 const int32* paddings_data = GetTensorData<int32>(paddings); in PadContext()
80 const int32* paddings_data = GetTensorData<int32>(op_context->paddings); in ResizeOutputTensor()
138 const int32* paddings_data = GetTensorData<int32>(op_context.paddings); in Eval()
162 GetTensorData<scalar>(op_context.input), &pad_value_copy, \ in Eval()
164 GetTensorData<scalar>(op_context.output)) in Eval()
169 : *GetTensorData<float>(op_context.constant_values); in Eval()
201 pad_value = *GetTensorData<uint8_t>(op_context.constant_values); in Eval()
234 pad_value = *GetTensorData<int8_t>(op_context.constant_values); in Eval()
246 : *GetTensorData<int32_t>(op_context.constant_values); in Eval()
257 : *GetTensorData<int64_t>(op_context.constant_values); in Eval()
Dfloor_div.cc90 const T* denominator_data = GetTensorData<T>(input2); in EvalImpl()
101 GetTensorShape(input1), GetTensorData<T>(input1), in EvalImpl()
103 GetTensorData<T>(output), FloorDiv<T>); in EvalImpl()
106 GetTensorShape(input1), GetTensorData<T>(input1), in EvalImpl()
107 GetTensorShape(input2), GetTensorData<T>(input2), in EvalImpl()
108 GetTensorShape(output), GetTensorData<T>(output), FloorDiv<T>); in EvalImpl()
Dl2norm.cc82 GetTensorData<float>(input), GetTensorShape(output), \ in Eval()
83 GetTensorData<float>(output)) in Eval()
97 GetTensorData<uint8>(input), GetTensorShape(output), \ in Eval()
98 GetTensorData<uint8>(output)) in Eval()
116 depth, GetTensorData<int8>(input), in Eval()
117 GetTensorData<int8>(output)); in Eval()
Dpooling.cc140 GetTensorData<float>(input), GetTensorShape(output), \ in AverageEvalFloat()
141 GetTensorData<float>(output)) in AverageEvalFloat()
170 GetTensorData<uint8_t>(input), GetTensorShape(output), \ in AverageEvalQuantizedUint8()
171 GetTensorData<uint8_t>(output)) in AverageEvalQuantizedUint8()
197 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in AverageEvalQuantizedInt8()
198 GetTensorShape(output), GetTensorData<int8_t>(output)); in AverageEvalQuantizedInt8()
218 type::MaxPool(op_params, GetTensorShape(input), GetTensorData<float>(input), \ in MaxEvalFloat()
219 GetTensorShape(output), GetTensorData<float>(output)) in MaxEvalFloat()
247 GetTensorData<uint8_t>(input), GetTensorShape(output), \ in MaxEvalQuantizedUInt8()
248 GetTensorData<uint8_t>(output)) in MaxEvalQuantizedUInt8()
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dfully_connected.cc102 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), \ in EvalQuantized()
103 GetTensorShape(filter), GetTensorData<uint8_t>(filter), \ in EvalQuantized()
104 GetTensorShape(bias), GetTensorData<int32_t>(bias), \ in EvalQuantized()
105 GetTensorShape(output), GetTensorData<output_data_type>(output), \ in EvalQuantized()
135 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
136 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat()
137 GetTensorShape(bias), GetTensorData<float>(bias), GetTensorShape(output), in EvalFloat()
138 GetTensorData<float>(output)); in EvalFloat()
Dsoftmax.cc101 GetTensorData<uint8_t>(input), shape, in Softmax1DQuantized()
102 GetTensorData<uint8_t>(output)); in Softmax1DQuantized()
120 GetTensorData<uint8_t>(input), shape, in Softmax2DQuantized()
121 GetTensorData<uint8_t>(output)); in Softmax2DQuantized()
130 op_params, GetTensorShape(input), GetTensorData<float>(input), in Softmax4DFloat()
131 GetTensorShape(output), GetTensorData<float>(output)); in Softmax4DFloat()
141 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in Softmax4DQuantized()
142 GetTensorShape(output), GetTensorData<uint8_t>(output)); in Softmax4DQuantized()
Ddepthwise_conv.cc116 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
117 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat()
118 GetTensorShape(bias), GetTensorData<float>(bias), GetTensorShape(output), in EvalFloat()
119 GetTensorData<float>(output)); in EvalFloat()
150 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
151 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
152 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized()
153 GetTensorShape(output), GetTensorData<uint8_t>(output)); in EvalQuantized()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_test.cc95 std::vector<T> GetTensorData(const tensorflow::Tensor& t) { in GetTensorData() function
113 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()
131 EXPECT_THAT(GetTensorData<string>(buffer_map.GetTensor(0)), in TEST()
151 EXPECT_THAT(GetTensorData<int>(buffer_map.GetTensor(0)), in TEST()
165 EXPECT_THAT(GetTensorData<string>(buffer_map.GetTensor(0)), in TEST()
176 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()
194 EXPECT_THAT(GetTensorData<int>(buffer_map.GetTensor(0)), in TEST()
209 EXPECT_THAT(GetTensorData<int>(buffer_map.GetTensor(0)), in TEST()
223 EXPECT_THAT(GetTensorData<float>(buffer_map.GetTensor(0)), in TEST()

123