Home
last modified time | relevance | path

Searched refs:GetTensorShape (Results 1 – 25 of 104) sorted by relevance

12345

/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()
425 GetTensorShape(output), in TanhEval()
428 reference_ops::Tanh(GetTensorShape(input), GetTensorData<float>(input), in TanhEval()
429 GetTensorShape(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.cc93 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()
111 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in Eval()
[all …]
Dreverse.cc78 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()
103 GetTensorShape(output), GetTensorData<int64_t>(output)); in Eval()
Dconv.cc461 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()
491 GetTensorShape(output), GetTensorData<uint8_t>(output), in EvalQuantized()
492 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), gemm_context); in EvalQuantized()
[all …]
Dmul.cc119 type::opname(op_params, GetTensorShape(input1), \ in EvalMul()
120 GetTensorData<data_type>(input1), GetTensorShape(input2), \ in EvalMul()
121 GetTensorData<data_type>(input2), GetTensorShape(output), \ in EvalMul()
172 GetTensorShape(input1), GetTensorShape(input2), &op_params); in EvalQuantized()
174 type::opname(op_params, GetTensorShape(input1), \ in EvalQuantized()
175 GetTensorData<dtype>(input1), GetTensorShape(input2), \ in EvalQuantized()
176 GetTensorData<dtype>(input2), GetTensorShape(output), \ in EvalQuantized()
205 type::opname(op_params, GetTensorShape(input1), \ in EvalQuantized()
206 GetTensorData<int16_t>(input1), GetTensorShape(input2), \ in EvalQuantized()
207 GetTensorData<int16_t>(input2), GetTensorShape(output), \ in EvalQuantized()
[all …]
Delementwise_test.cc65 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1})); in TEST()
74 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1})); in TEST()
83 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1})); in TEST()
105 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1})); in TEST()
114 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1})); in TEST()
123 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1})); in TEST()
132 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({1, 1, 4, 1})); in TEST()
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()
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 …]
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()
260 depthwise_conv(op_params, GetTensorShape(input), in EvalQuantized()
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()
287 data->per_channel_output_shift.data(), GetTensorShape(input), in EvalQuantizedPerChannel()
288 GetTensorData<int8>(input), GetTensorShape(filter), in EvalQuantizedPerChannel()
[all …]
Dcomparisons.cc91 op_params, GetTensorShape(input1), \
92 GetTensorData<input_dtype>(input1), GetTensorShape(input2), \
93 GetTensorData<input_dtype>(input2), GetTensorShape(output), \
97 op_params, GetTensorShape(input1), \
98 GetTensorData<input_dtype>(input1), GetTensorShape(input2), \
99 GetTensorData<input_dtype>(input2), GetTensorShape(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()
Dadd.cc186 type::opname(op_params, GetTensorShape(input1), \ in EvalAdd()
187 GetTensorData<data_type>(input1), GetTensorShape(input2), \ in EvalAdd()
188 GetTensorData<data_type>(input2), GetTensorShape(output), \ in EvalAdd()
243 GetTensorShape(input1), GetTensorShape(input2), &op_params); in EvalAddQuantized()
245 type::opname(op_params, GetTensorShape(input1), \ in EvalAddQuantized()
246 GetTensorData<dtype>(input1), GetTensorShape(input2), \ in EvalAddQuantized()
247 GetTensorData<dtype>(input2), GetTensorShape(output), \ in EvalAddQuantized()
281 type::opname(op_params, GetTensorShape(input1), \ in EvalAddQuantized()
282 GetTensorData<int16_t>(input1), GetTensorShape(input2), \ in EvalAddQuantized()
283 GetTensorData<int16_t>(input2), GetTensorShape(output), \ in EvalAddQuantized()
Dl2norm.cc81 type::L2Normalization(op_params, GetTensorShape(input), \ in Eval()
82 GetTensorData<float>(input), GetTensorShape(output), \ in Eval()
96 type::L2Normalization(op_params, GetTensorShape(input), \ in Eval()
97 GetTensorData<uint8>(input), GetTensorShape(output), \ in Eval()
108 const auto input_shape = GetTensorShape(input); in Eval()
109 const auto output_shape = GetTensorShape(output); in Eval()
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()
Dfloor_div.cc101 GetTensorShape(input1), GetTensorData<T>(input1), in EvalImpl()
102 GetTensorShape(input2), denominator_data, GetTensorShape(output), 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()
Dpooling.cc139 type::AveragePool(op_params, GetTensorShape(input), \ in AverageEvalFloat()
140 GetTensorData<float>(input), GetTensorShape(output), \ in AverageEvalFloat()
169 type::AveragePool(op_params, GetTensorShape(input), \ in AverageEvalQuantizedUint8()
170 GetTensorData<uint8_t>(input), GetTensorShape(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()
246 type::MaxPool(op_params, GetTensorShape(input), \ in MaxEvalQuantizedUInt8()
247 GetTensorData<uint8_t>(input), GetTensorShape(output), \ in MaxEvalQuantizedUInt8()
[all …]
Dtranspose_conv.cc222 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()
Dfloor_mod.cc122 GetTensorShape(input1), GetTensorData<T>(input1), in EvalImpl()
123 GetTensorShape(input2), denominator_data, GetTensorShape(output), in EvalImpl()
127 GetTensorShape(input1), GetTensorData<T>(input1), in EvalImpl()
128 GetTensorShape(input2), GetTensorData<T>(input2), in EvalImpl()
129 GetTensorShape(output), GetTensorData<T>(output), FloorMod<T>); in EvalImpl()
Dfloor.cc52 optimized_ops::Floor(GetTensorShape(input), GetTensorData<float>(input), in Eval()
53 GetTensorShape(output), GetTensorData<float>(output)); in Eval()
55 reference_ops::Floor(GetTensorShape(input), GetTensorData<float>(input), in Eval()
56 GetTensorShape(output), GetTensorData<float>(output)); in Eval()
Dfill_test.cc55 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({2, 3})); in TEST()
66 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({2, 4})); in TEST()
76 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({2, 2, 2})); in TEST()
84 EXPECT_THAT(m.GetTensorShape(m.output()), IsEmpty()); in TEST()
Ddequantize.cc87 optimized_ops::Dequantize(op_params, GetTensorShape(op_context.input), in Eval()
89 GetTensorShape(op_context.output), in Eval()
94 op_params, GetTensorShape(op_context.input), in Eval()
96 GetTensorShape(op_context.output), in Eval()
/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()
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()
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/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc307 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
308 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat()
309 GetTensorShape(bias), GetTensorData<float>(bias), GetTensorShape(output), in EvalFloat()
341 const int filter_width = GetTensorShape(filter).Dims(2); in EvalQuantized()
342 const int input_depth = GetTensorShape(input).Dims(3); in EvalQuantized()
343 const int output_depth = GetTensorShape(filter).Dims(3); in EvalQuantized()
344 const int filter_height = GetTensorShape(filter).Dims(1); in EvalQuantized()
373 context, op_params, GetTensorShape(input), in EvalQuantized()
374 GetTensorData<uint8_t>(input), GetTensorShape(filter), in EvalQuantized()
375 GetTensorData<uint8_t>(filter), GetTensorShape(bias), in EvalQuantized()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_test.cc25 RuntimeShape d = GetTensorShape({2, 3, 4, 5}); in TEST()
32 RuntimeShape d = GetTensorShape({3, 4, 5}); in TEST()
39 RuntimeShape d = GetTensorShape({4, 5}); in TEST()
46 RuntimeShape d = GetTensorShape({5}); in TEST()

12345