Home
last modified time | relevance | path

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

12345678910

/external/tensorflow/tensorflow/lite/kernels/
Ddequantize.h51 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in DequantizeImpl()
52 GetTensorShape(output), GetTensorData<float>(output)); in DequantizeImpl()
55 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in DequantizeImpl()
56 GetTensorShape(output), GetTensorData<float>(output)); in DequantizeImpl()
62 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in DequantizeImpl()
63 GetTensorShape(output), GetTensorData<float>(output)); in DequantizeImpl()
66 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in DequantizeImpl()
67 GetTensorShape(output), GetTensorData<float>(output)); in DequantizeImpl()
73 op_params, GetTensorShape(input), GetTensorData<int16_t>(input), in DequantizeImpl()
74 GetTensorShape(output), GetTensorData<float>(output)); in DequantizeImpl()
[all …]
Dactivations.cc156 MatchingFlatSize(GetTensorShape(input), GetTensorShape(output)); in EvalUsingLookupTable()
206 optimized_ops::ReluX(params, GetTensorShape(input), GetTensorData<T>(input), in QuantizedReluX()
207 GetTensorShape(output), GetTensorData<T>(output)); in QuantizedReluX()
750 optimized_ops::Relu(GetTensorShape(input), GetTensorData<float>(input), in ReluEval()
751 GetTensorShape(output), GetTensorData<float>(output)); in ReluEval()
785 optimized_ops::Relu1(GetTensorShape(input), GetTensorData<float>(input), in Relu1Eval()
786 GetTensorShape(output), in Relu1Eval()
819 GetTensorShape(input), GetTensorData<float>(input), in HardSwishEval()
820 GetTensorShape(output), GetTensorData<float>(output)); in HardSwishEval()
823 GetTensorShape(input), GetTensorData<float>(input), in HardSwishEval()
[all …]
Dmaximum_minimum.cc97 GetTensorShape(op_context.input1), in TFLiteOperation()
99 GetTensorShape(op_context.input2), in TFLiteOperation()
101 GetTensorShape(op_context.output), in TFLiteOperation()
112 GetTensorShape(op_context.input1), GetTensorShape(op_context.input2), in TFLiteOperation()
116 op_params, GetTensorShape(op_context.input1), in TFLiteOperation()
118 GetTensorShape(op_context.input2), in TFLiteOperation()
120 GetTensorShape(op_context.output), 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()
[all …]
Dfully_connected.cc444 const auto& input_shape = GetTensorShape(input); in EvalHybridImpl()
445 const auto& output_shape = GetTensorShape(output); in EvalHybridImpl()
446 const auto& filter_shape = GetTensorShape(filter); in EvalHybridImpl()
587 const auto& output_shape = GetTensorShape(output); in EvalHybrid()
650 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in FullyConnectedInt8()
651 GetTensorShape(filter), GetTensorData<int8_t>(filter), in FullyConnectedInt8()
652 GetTensorShape(bias), GetTensorData<int32_t>(bias), in FullyConnectedInt8()
653 GetTensorShape(output), GetTensorData<int8_t>(output)); in FullyConnectedInt8()
656 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in FullyConnectedInt8()
657 GetTensorShape(filter), GetTensorData<int8_t>(filter), in FullyConnectedInt8()
[all …]
Dresize_nearest_neighbor.cc111 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()
129 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in Eval()
[all …]
Dreverse.cc94 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()
120 GetTensorShape(output), GetTensorData<int64_t>(output)); in Eval()
[all …]
Dtranspose_conv.cc179 const RuntimeShape& input_shape = GetTensorShape(input); in ResizeCol2ImTensor()
180 const RuntimeShape& weights_shape = GetTensorShape(weights); in ResizeCol2ImTensor()
194 const RuntimeShape& input_shape = GetTensorShape(weights); in ResizeAndTransposeWeights()
216 GetTensorShape(transposed_weights), in ResizeAndTransposeWeights()
221 GetTensorShape(transposed_weights), in ResizeAndTransposeWeights()
227 GetTensorShape(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()
415 GetTensorShape(output), GetTensorData<float>(output), in EvalFloat()
[all …]
Ddepthwise_conv.cc317 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
318 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat()
319 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat()
320 GetTensorShape(output), GetTensorData<float>(output)); in EvalFloat()
323 op_params, GetTensorShape(input), GetTensorData<float>(input), in EvalFloat()
324 GetTensorShape(filter), GetTensorData<float>(filter), in EvalFloat()
325 GetTensorShape(bias), GetTensorData<float>(bias), in EvalFloat()
326 GetTensorShape(output), GetTensorData<float>(output), in EvalFloat()
361 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
362 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
[all …]
Dconv.cc643 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
644 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
645 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized()
646 GetTensorShape(output), GetTensorData<uint8_t>(output), in EvalQuantized()
647 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), in EvalQuantized()
656 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in EvalQuantized()
657 GetTensorShape(filter), GetTensorData<uint8_t>(filter), in EvalQuantized()
658 GetTensorShape(bias), GetTensorData<int32_t>(bias), in EvalQuantized()
659 GetTensorShape(output), GetTensorData<uint8_t>(output), in EvalQuantized()
660 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), in EvalQuantized()
[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 …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dconv.cc62 ConvParamsFloat(params, data), tflite::micro::GetTensorShape(input), in Eval()
64 tflite::micro::GetTensorShape(filter), in Eval()
66 tflite::micro::GetTensorShape(bias), in Eval()
68 tflite::micro::GetTensorShape(output), in Eval()
70 tflite::micro::GetTensorShape(nullptr), nullptr); in Eval()
76 data.per_channel_output_shift, tflite::micro::GetTensorShape(input), in Eval()
78 tflite::micro::GetTensorShape(filter), in Eval()
80 tflite::micro::GetTensorShape(bias), in Eval()
82 tflite::micro::GetTensorShape(output), in Eval()
88 tflite::micro::GetTensorShape(input), in Eval()
[all …]
Dmul.cc104 tflite::micro::GetTensorShape(input1), in EvalQuantized()
105 tflite::micro::GetTensorShape(input2), &op_params); in EvalQuantized()
110 op_params, tflite::micro::GetTensorShape(input1), in EvalQuantized()
112 tflite::micro::GetTensorShape(input2), in EvalQuantized()
114 tflite::micro::GetTensorShape(output), in EvalQuantized()
118 tflite::micro::GetTensorShape(input1), in EvalQuantized()
120 tflite::micro::GetTensorShape(input2), in EvalQuantized()
122 tflite::micro::GetTensorShape(output), in EvalQuantized()
128 op_params, tflite::micro::GetTensorShape(input1), in EvalQuantized()
130 tflite::micro::GetTensorShape(input2), in EvalQuantized()
[all …]
Dsub.cc132 op_params, tflite::micro::GetTensorShape(input1), in EvalSub()
134 tflite::micro::GetTensorShape(input2), in EvalSub()
136 tflite::micro::GetTensorShape(output), in EvalSub()
140 op_params, tflite::micro::GetTensorShape(input1), in EvalSub()
142 tflite::micro::GetTensorShape(input2), in EvalSub()
144 tflite::micro::GetTensorShape(output), in EvalSub()
169 tflite::micro::GetTensorShape(input1), in EvalSubQuantized()
170 tflite::micro::GetTensorShape(input2), &op_params); in EvalSubQuantized()
175 op_params, tflite::micro::GetTensorShape(input1), in EvalSubQuantized()
177 tflite::micro::GetTensorShape(input2), in EvalSubQuantized()
[all …]
Dadd.cc115 op_params, tflite::micro::GetTensorShape(input1), in EvalAdd()
117 tflite::micro::GetTensorShape(input2), in EvalAdd()
119 tflite::micro::GetTensorShape(output), in EvalAdd()
122 reference_ops::Add(op_params, tflite::micro::GetTensorShape(input1), in EvalAdd()
124 tflite::micro::GetTensorShape(input2), in EvalAdd()
126 tflite::micro::GetTensorShape(output), in EvalAdd()
151 tflite::micro::GetTensorShape(input1), in EvalAddQuantized()
152 tflite::micro::GetTensorShape(input2), &op_params); in EvalAddQuantized()
156 op_params, tflite::micro::GetTensorShape(input1), in EvalAddQuantized()
158 tflite::micro::GetTensorShape(input2), in EvalAddQuantized()
[all …]
Dsoftmax.cc35 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
37 tflite::micro::GetTensorShape(output), in SoftmaxQuantized()
42 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
44 tflite::micro::GetTensorShape(output), in SoftmaxQuantized()
48 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
50 tflite::micro::GetTensorShape(output), in SoftmaxQuantized()
55 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
57 tflite::micro::GetTensorShape(output), in SoftmaxQuantized()
72 op_data, tflite::micro::GetTensorShape(input), in SoftmaxEval()
74 tflite::micro::GetTensorShape(output), in SoftmaxEval()
Dfully_connected.cc87 tflite::micro::GetTensorShape(input), in Eval()
89 tflite::micro::GetTensorShape(filter), in Eval()
91 tflite::micro::GetTensorShape(bias), in Eval()
93 tflite::micro::GetTensorShape(output), in Eval()
101 tflite::micro::GetTensorShape(input), in Eval()
103 tflite::micro::GetTensorShape(filter), in Eval()
105 tflite::micro::GetTensorShape(bias), in Eval()
107 tflite::micro::GetTensorShape(output), in Eval()
115 tflite::micro::GetTensorShape(input), in Eval()
117 tflite::micro::GetTensorShape(filter), in Eval()
[all …]
Dpad.cc74 TF_LITE_ENSURE_EQ(context, GetTensorShape(paddings).FlatSize(), in Prepare()
91 const int paddings_total = GetTensorShape(paddings).FlatSize(); in Prepare()
156 data->params, tflite::micro::GetTensorShape(input), in Eval()
158 tflite::micro::GetTensorShape(output), in Eval()
161 reference_ops::Pad(data->params, tflite::micro::GetTensorShape(input), in Eval()
163 &pad_value, tflite::micro::GetTensorShape(output), in Eval()
176 data->params, tflite::micro::GetTensorShape(input), in Eval()
178 tflite::micro::GetTensorShape(output), in Eval()
181 reference_ops::Pad(data->params, tflite::micro::GetTensorShape(input), in Eval()
183 &pad_value, tflite::micro::GetTensorShape(output), in Eval()
[all …]
Dresize_nearest_neighbor.cc75 op_params, tflite::micro::GetTensorShape(input), in Eval()
77 tflite::micro::GetTensorShape(size), in Eval()
79 tflite::micro::GetTensorShape(output), in Eval()
83 op_params, tflite::micro::GetTensorShape(input), in Eval()
85 tflite::micro::GetTensorShape(size), in Eval()
87 tflite::micro::GetTensorShape(output), in Eval()
91 op_params, tflite::micro::GetTensorShape(input), in Eval()
93 tflite::micro::GetTensorShape(size), in Eval()
95 tflite::micro::GetTensorShape(output), in Eval()
Dbatch_to_space_nd.cc74 tflite::micro::GetTensorShape(input), in Eval()
76 tflite::micro::GetTensorShape(block_shape), in Eval()
78 tflite::micro::GetTensorShape(crops), in Eval()
80 tflite::micro::GetTensorShape(output), in Eval()
85 tflite::micro::GetTensorShape(input), in Eval()
87 tflite::micro::GetTensorShape(block_shape), in Eval()
89 tflite::micro::GetTensorShape(crops), in Eval()
91 tflite::micro::GetTensorShape(output), in Eval()
Dspace_to_batch_nd.cc84 params, tflite::micro::GetTensorShape(input), in Eval()
86 tflite::micro::GetTensorShape(block_shape), in Eval()
88 tflite::micro::GetTensorShape(crops), in Eval()
90 tflite::micro::GetTensorShape(output), in Eval()
95 params, tflite::micro::GetTensorShape(input), in Eval()
97 tflite::micro::GetTensorShape(block_shape), in Eval()
99 tflite::micro::GetTensorShape(crops), in Eval()
101 tflite::micro::GetTensorShape(output), in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsoftmax.cc35 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
37 tflite::micro::GetTensorShape(output), in SoftmaxQuantized()
42 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
44 tflite::micro::GetTensorShape(output), in SoftmaxQuantized()
47 const auto input_shape = tflite::micro::GetTensorShape(input); in SoftmaxQuantized()
48 const auto output_shape = tflite::micro::GetTensorShape(output); in SoftmaxQuantized()
62 op_data, tflite::micro::GetTensorShape(input), in SoftmaxQuantized()
64 tflite::micro::GetTensorShape(output), in SoftmaxQuantized()
80 data, tflite::micro::GetTensorShape(input), in SoftmaxEval()
82 tflite::micro::GetTensorShape(output), in SoftmaxEval()
Dconv.cc59 RuntimeShape input_shape = GetTensorShape(input); in Prepare()
60 RuntimeShape output_shape = GetTensorShape(output); in Prepare()
157 RuntimeShape filter_shape = tflite::micro::GetTensorShape(filter); in EvalQuantizedPerChannel()
158 RuntimeShape input_shape = tflite::micro::GetTensorShape(input); in EvalQuantizedPerChannel()
159 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in EvalQuantizedPerChannel()
160 RuntimeShape bias_shape = tflite::micro::GetTensorShape(bias); in EvalQuantizedPerChannel()
230 tflite::micro::GetTensorShape(input), in EvalQuantizedPerChannel()
232 tflite::micro::GetTensorShape(filter), in EvalQuantizedPerChannel()
234 tflite::micro::GetTensorShape(bias), in EvalQuantizedPerChannel()
236 tflite::micro::GetTensorShape(output), in EvalQuantizedPerChannel()
[all …]
Dfully_connected.cc75 RuntimeShape filter_shape = GetTensorShape(filter); in Prepare()
76 RuntimeShape output_shape = GetTensorShape(output); in Prepare()
108 const RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in EvalQuantizedInt8()
112 const RuntimeShape filter_shape = tflite::micro::GetTensorShape(filter); in EvalQuantizedInt8()
115 const RuntimeShape input_shape = tflite::micro::GetTensorShape(input); in EvalQuantizedInt8()
172 tflite::micro::GetTensorShape(input), in EvalQuantizedInt8()
174 tflite::micro::GetTensorShape(filter), in EvalQuantizedInt8()
176 tflite::micro::GetTensorShape(bias), in EvalQuantizedInt8()
178 tflite::micro::GetTensorShape(output), in EvalQuantizedInt8()
206 tflite::micro::GetTensorShape(input), in Eval()
[all …]
Dmul.cc116 tflite::micro::GetTensorShape(input1), in EvalQuantized()
117 tflite::micro::GetTensorShape(input2), &op_params); in EvalQuantized()
120 type::opname(op_params, tflite::micro::GetTensorShape(input1), \ in EvalQuantized()
122 tflite::micro::GetTensorShape(input2), \ in EvalQuantized()
124 tflite::micro::GetTensorShape(output), \ in EvalQuantized()
139 MatchingElementsSize(tflite::micro::GetTensorShape(input1), in EvalQuantized()
140 tflite::micro::GetTensorShape(input2), in EvalQuantized()
141 tflite::micro::GetTensorShape(output))); in EvalQuantized()
164 tflite::micro::GetTensorShape(input1), in EvalFloat()
165 tflite::micro::GetTensorShape(input2), &op_params); in EvalFloat()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util.cc47 GetTensorShape(&original_tflite_tensor), in DequantizeInput()
49 GetTensorShape(&original_tflite_tensor), in DequantizeInput()
53 GetTensorShape(&original_tflite_tensor), in DequantizeInput()
55 GetTensorShape(&original_tflite_tensor), in DequantizeInput()
75 GetTensorShape(&original_tflite_tensor), in QuantizeOutput()
77 GetTensorShape(&original_tflite_tensor), in QuantizeOutput()
81 GetTensorShape(&original_tflite_tensor), in QuantizeOutput()
83 GetTensorShape(&original_tflite_tensor), in QuantizeOutput()

12345678910