Home
last modified time | relevance | path

Searched refs:output_shape (Results 1 – 25 of 101) sorted by relevance

12345

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Dpooling.h31 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
36 const int output_height = output_shape.Dims(1); in AveragePool()
37 const int output_width = output_shape.Dims(2); in AveragePool()
92 const int output_height = output_shape.Dims(1); in AveragePool()
93 const int output_width = output_shape.Dims(2); in AveragePool()
145 const int output_height = output_shape.Dims(1); in L2Pool()
146 const int output_width = output_shape.Dims(2); in L2Pool()
199 const int output_height = output_shape.Dims(1); in MaxPool()
200 const int output_width = output_shape.Dims(2); in MaxPool()
253 const int output_height = output_shape.Dims(1); in MaxPool()
[all …]
A Dcomparisons.h62 MatchingFlatSize(input1_shape, input2_shape, output_shape); in ComparisonImpl()
110 const RuntimeShape output_shape; member
141 for (int b = 0; b < dims.output_shape.Dims(0); ++b) { in BroadcastComparison4DSlowImpl()
142 for (int y = 0; y < dims.output_shape.Dims(1); ++y) { in BroadcastComparison4DSlowImpl()
143 for (int x = 0; x < dims.output_shape.Dims(2); ++x) { in BroadcastComparison4DSlowImpl()
144 for (int c = 0; c < dims.output_shape.Dims(3); ++c) { in BroadcastComparison4DSlowImpl()
145 output_data[Offset(dims.output_shape, b, y, x, c)] = in BroadcastComparison4DSlowImpl()
186 for (int b = 0; b < dims.output_shape.Dims(0); ++b) { in BroadcastComparison4DSlowWithScaling()
187 for (int y = 0; y < dims.output_shape.Dims(1); ++y) { in BroadcastComparison4DSlowWithScaling()
188 for (int x = 0; x < dims.output_shape.Dims(2); ++x) { in BroadcastComparison4DSlowWithScaling()
[all …]
A Dtranspose_conv.h29 const float* bias_data, const RuntimeShape& output_shape, in TransposeConv() argument
37 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in TransposeConv()
41 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConv()
48 const int output_height = output_shape.Dims(1); in TransposeConv()
49 const int output_width = output_shape.Dims(2); in TransposeConv()
61 const int num_elements = output_shape.FlatSize(); in TransposeConv()
118 const int32_t* bias_data, const RuntimeShape& output_shape, in TransposeConv() argument
127 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in TransposeConv()
138 const int output_height = output_shape.Dims(1); in TransposeConv()
139 const int output_width = output_shape.Dims(2); in TransposeConv()
[all …]
A Dconcatenation.h31 const RuntimeShape& output_shape, in Concatenation() argument
35 const int concat_dimensions = output_shape.DimensionsCount(); in Concatenation()
43 MatchingDim(*input_shapes[i], j, output_shape, j); in Concatenation()
48 TFLITE_DCHECK_EQ(concat_size, output_shape.Dims(axis)); in Concatenation()
51 outer_size *= output_shape.Dims(i); in Concatenation()
57 base_inner_size *= output_shape.Dims(i); in Concatenation()
86 const int concat_dimensions = output_shape.DimensionsCount(); in ConcatenationWithScaling()
94 MatchingDim(*input_shapes[i], j, output_shape, j); in ConcatenationWithScaling()
99 TFLITE_DCHECK_EQ(concat_size, output_shape.Dims(axis)); in ConcatenationWithScaling()
102 outer_size *= output_shape.Dims(i); in ConcatenationWithScaling()
[all …]
A Dconv.h41 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
45 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv()
55 const int output_height = output_shape.Dims(1); in Conv()
56 const int output_width = output_shape.Dims(2); in Conv()
127 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
138 const int output_height = output_shape.Dims(1); in Conv()
139 const int output_width = output_shape.Dims(2); in Conv()
192 const RuntimeShape& output_shape, float* output_data, in HybridConvPerChannel() argument
207 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in HybridConvPerChannel()
218 const int output_height = output_shape.Dims(1); in HybridConvPerChannel()
[all …]
A Dpad.h40 const P* pad_value_ptr, const RuntimeShape& output_shape, in PadImpl() argument
45 RuntimeShape::ExtendedShape(PadKernelMaxDimensionCount(), output_shape); in PadImpl()
120 const P* pad_value_ptr, const RuntimeShape& output_shape, in Pad() argument
122 PadImpl(op_params, input_shape, input_data, pad_value_ptr, output_shape, in Pad()
130 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad() argument
134 output_shape, output_data); in Pad()
143 PadImpl(op_params, input_shape, input_data, pad_value_ptr, output_shape, in Pad()
151 const RuntimeShape& output_shape, T* output_data) { in PadImageStyle() argument
152 Pad(op_params, input_shape, input_data, pad_value_ptr, output_shape, in PadImageStyle()
160 const RuntimeShape& output_shape, in PadImageStyle() argument
[all …]
A Dbinary_function.h38 const RuntimeShape output_shape = in BroadcastBinaryFunction4DSlow() local
46 for (int b = 0; b < output_shape.Dims(0); ++b) { in BroadcastBinaryFunction4DSlow()
47 for (int y = 0; y < output_shape.Dims(1); ++y) { in BroadcastBinaryFunction4DSlow()
48 for (int x = 0; x < output_shape.Dims(2); ++x) { in BroadcastBinaryFunction4DSlow()
49 for (int c = 0; c < output_shape.Dims(3); ++c) { in BroadcastBinaryFunction4DSlow()
50 auto out_idx = Offset(output_shape, b, y, x, c); in BroadcastBinaryFunction4DSlow()
68 const RuntimeShape& output_shape, R* output_data, in BinaryFunction() argument
71 MatchingFlatSize(input1_shape, input2_shape, output_shape); in BinaryFunction()
A Dmul.h48 const RuntimeShape& output_shape, T* output_data) { in Mul() argument
54 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
69 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
79 const RuntimeShape& output_shape, in BroadcastMul4DSlow() argument
86 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastMul4DSlow()
129 const RuntimeShape output_shape = in BroadcastMul4DSlow() local
148 for (int b = 0; b < output_shape.Dims(0); ++b) { in BroadcastMul4DSlow()
149 for (int y = 0; y < output_shape.Dims(1); ++y) { in BroadcastMul4DSlow()
150 for (int x = 0; x < output_shape.Dims(2); ++x) { in BroadcastMul4DSlow()
151 for (int c = 0; c < output_shape.Dims(3); ++c) { in BroadcastMul4DSlow()
[all …]
A Dlogistic.h31 const RuntimeShape& output_shape, float* output_data) { in Logistic() argument
35 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic()
62 const float* input_data, const RuntimeShape& output_shape, in Logistic() argument
65 Logistic(input_shape, input_data, output_shape, output_data); in Logistic()
70 const RuntimeShape& output_shape, int16_t* output_data) { in Logistic() argument
71 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic()
95 const RuntimeShape& output_shape, int8_t* output_data, in Logistic() argument
100 const int flat_size = MatchingFlatSize(input_shape, output_shape); in Logistic()
A Dreference_ops.h328 outer_size *= output_shape.Dims(i); in Pack()
332 copy_size *= output_shape.Dims(i); in Pack()
394 outer_size *= output_shape.Dims(i); in PackWithScaling()
398 copy_size *= output_shape.Dims(i); in PackWithScaling()
1037 int output_flat_size = output_shape.FlatSize(); in ScatterNd()
1062 const RuntimeShape& output_shape, in Slice() argument
1235 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastSelect4DSlow()
1312 const RuntimeShape output_shape = in SparseToDense() local
1317 const int num_elements = output_shape.FlatSize(); in SparseToDense()
1366 const RuntimeShape output_shape = in BroadcastPow4DSlow() local
[all …]
A Dfully_connected.h30 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
39 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnected()
41 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnected()
66 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
76 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
84 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected()
114 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
131 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected()
168 const int32_t* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected() argument
178 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in ShuffledFullyConnected()
[all …]
A Dsub.h37 const RuntimeShape& output_shape, in SubNonBroadcast() argument
53 const RuntimeShape& output_shape, in SubNonBroadcast() argument
74 const RuntimeShape& output_shape, in BroadcastSubSlow() argument
79 TFLITE_DCHECK_LE(output_shape.DimensionsCount(), N); in BroadcastSubSlow()
114 const RuntimeShape& output_shape, in BroadcastSubSlow() argument
119 TFLITE_DCHECK_LE(output_shape.DimensionsCount(), N); in BroadcastSubSlow()
171 const RuntimeShape& output_shape, in BroadcastSubSlow() argument
176 TFLITE_DCHECK_LE(output_shape.DimensionsCount(), N); in BroadcastSubSlow()
269 TFLITE_DCHECK_LE(output_shape.DimensionsCount(), N); in BroadcastSubSlow()
306 TFLITE_DCHECK_LE(output_shape.DimensionsCount(), N); in BroadcastSubSlow()
[all …]
A Ddepthwiseconv_uint8.h96 const int32_t* bias_data, const RuntimeShape& output_shape, in Run()
114 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Run()
117 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Run()
124 const int output_height = output_shape.Dims(1); in Run()
125 const int output_width = output_shape.Dims(2); in Run()
180 const int32_t* bias_data, const RuntimeShape& output_shape, in RunPerChannel()
201 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in RunPerChannel()
211 const int output_height = output_shape.Dims(1); in RunPerChannel()
212 const int output_width = output_shape.Dims(2); in RunPerChannel()
268 output_data[Offset(output_shape, batch, out_y, out_x, in RunPerChannel()
[all …]
A Dstring_comparisons.h45 const RuntimeShape& output_shape, in ComparisonStringImpl() argument
48 MatchingFlatSize(input1_shape, input2_shape, output_shape); in ComparisonStringImpl()
66 for (int b = 0; b < dims.output_shape.Dims(0); ++b) { in BroadcastComparison4DSlowStringImpl()
67 for (int y = 0; y < dims.output_shape.Dims(1); ++y) { in BroadcastComparison4DSlowStringImpl()
68 for (int x = 0; x < dims.output_shape.Dims(2); ++x) { in BroadcastComparison4DSlowStringImpl()
69 for (int c = 0; c < dims.output_shape.Dims(3); ++c) { in BroadcastComparison4DSlowStringImpl()
74 output_data[Offset(dims.output_shape, b, y, x, c)] = F(lhs, rhs); in BroadcastComparison4DSlowStringImpl()
A Dconv3d.h27 const float* bias_data, const RuntimeShape& output_shape, in Conv3D() argument
31 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 5); in Conv3D()
33 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv3D()
35 const int output_num_channels = MatchingDim(filter_shape, 4, output_shape, 4); in Conv3D()
47 const int output_width = output_shape.Dims(3); in Conv3D()
48 const int output_height = output_shape.Dims(2); in Conv3D()
49 const int output_depth = output_shape.Dims(1); in Conv3D()
99 output_data[Offset(output_shape, batch, out_d, out_y, out_x, in Conv3D()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/integer_ops/
A Dpooling.h31 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
36 const int output_height = output_shape.Dims(1); in AveragePool()
37 const int output_width = output_shape.Dims(2); in AveragePool()
92 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in MaxPool()
97 const int output_height = output_shape.Dims(1); in MaxPool()
98 const int output_width = output_shape.Dims(2); in MaxPool()
147 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
152 const int output_height = output_shape.Dims(1); in AveragePool()
153 const int output_width = output_shape.Dims(2); in AveragePool()
213 const int output_height = output_shape.Dims(1); in MaxPool()
[all …]
A Dtranspose_conv.h29 const int32_t* bias_data, const RuntimeShape& output_shape, in TransposeConv() argument
38 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in TransposeConv()
42 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConv()
52 const int output_height = output_shape.Dims(1); in TransposeConv()
53 const int output_width = output_shape.Dims(2); in TransposeConv()
60 const int num_elements = output_shape.FlatSize(); in TransposeConv()
137 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in TransposeConv()
141 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConv()
151 const int output_height = output_shape.Dims(1); in TransposeConv()
152 const int output_width = output_shape.Dims(2); in TransposeConv()
[all …]
A Ddepthwise_conv.h27 const int32_t* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel() argument
46 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
56 const int output_height = output_shape.Dims(1); in DepthwiseConvPerChannel()
57 const int output_width = output_shape.Dims(2); in DepthwiseConvPerChannel()
143 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
153 const int output_height = output_shape.Dims(1); in DepthwiseConvPerChannel()
154 const int output_width = output_shape.Dims(2); in DepthwiseConvPerChannel()
213 const RuntimeShape& output_shape, float* output_data, in DepthwiseConvHybridPerChannel() argument
227 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvHybridPerChannel()
236 const int output_height = output_shape.Dims(1); in DepthwiseConvHybridPerChannel()
[all …]
A Dconv.h29 const int32_t* bias_data, const RuntimeShape& output_shape, in ConvPerChannel() argument
49 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in ConvPerChannel()
50 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in ConvPerChannel()
52 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel()
62 const int output_height = output_shape.Dims(1); in ConvPerChannel()
63 const int output_width = output_shape.Dims(2); in ConvPerChannel()
134 const std::int64_t* bias_data, const RuntimeShape& output_shape, in ConvPerChannel() argument
152 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in ConvPerChannel()
153 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in ConvPerChannel()
165 const int output_height = output_shape.Dims(1); in ConvPerChannel()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dcomparisons.cc50 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in EqualEval() local
60 tflite::micro::GetTensorData<bool>(input2), output_shape, in EqualEval()
65 tflite::micro::GetTensorData<bool>(input2), output_shape, in EqualEval()
73 tflite::micro::GetTensorData<float>(input2), output_shape, in EqualEval()
78 tflite::micro::GetTensorData<float>(input2), output_shape, in EqualEval()
155 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in NotEqualEval() local
165 tflite::micro::GetTensorData<bool>(input2), output_shape, in NotEqualEval()
259 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in GreaterEval() local
350 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in GreaterEqualEval() local
441 RuntimeShape output_shape = tflite::micro::GetTensorShape(output); in LessEval() local
[all …]
A Dreshape.cc43 TfLiteIntArray* output_shape = output->dims; in ReshapeOutput() local
46 output_shape->size == 1 && output_shape->data[0] == 0) { in ReshapeOutput()
50 output_shape->size = 0; in ReshapeOutput()
55 for (int i = 0; i < output_shape->size; ++i) { in ReshapeOutput()
56 int value = output_shape->data[i]; in ReshapeOutput()
65 output_shape->data[stretch_dim] = num_input_elements / num_output_elements; in ReshapeOutput()
66 num_output_elements *= output_shape->data[stretch_dim]; in ReshapeOutput()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/sparse_ops/
A Dfully_connected.h35 const RuntimeShape& output_shape, float* output_data) { in FullyConnectedSparseWeight() argument
41 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight()
42 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnectedSparseWeight()
89 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnectedSparseWeight1x4Impl()
131 output_shape(output_shape), in FullyConnectedSparseWeight1x4Task()
140 bias_shape, bias_data, output_shape, output_data, thread_start, in Run()
153 const RuntimeShape& output_shape; member
169 const RuntimeShape& output_shape, float* output_data, in FullyConnectedSparseWeight1x4() argument
171 const int output_elements = output_shape.FlatSize(); in FullyConnectedSparseWeight1x4()
176 FlatSizeSkipDim(output_shape, output_shape.DimensionsCount() - 1); in FullyConnectedSparseWeight1x4()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/
A Ddepthwise_conv_hybrid.h45 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConvHybridGeneral() argument
65 const int output_rows = output_shape.Dims(1); in DepthwiseConvHybridGeneral()
66 const int output_width = output_shape.Dims(2); in DepthwiseConvHybridGeneral()
264 const RuntimeShape& output_shape, float* output_data, in DepthwiseConvHybridWithRounding() argument
275 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvHybridWithRounding()
296 depth_multiplier, output_shape, 0, nullptr)) { in DepthwiseConvHybridWithRounding()
358 output_shape(output_shape), in DepthwiseConvHybridWorkerTask()
383 const RuntimeShape& output_shape; member
403 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvHybridPerChannel()
405 const int output_batches = output_shape.Dims(0); in DepthwiseConvHybridPerChannel()
[all …]
A Dmean.h35 const int output_batch = output_shape.Dims(0); in MeanImpl()
36 const int output_height = output_shape.Dims(2); in MeanImpl()
37 const int output_width = output_shape.Dims(2); in MeanImpl()
137 output_data[Offset(output_shape, out_b, 0, 0, out_d)] = in MeanImpl()
155 output_shape(output_shape), in MeanWorkerTask()
172 const RuntimeShape& output_shape; member
191 const RuntimeShape output_shape = in Mean() local
193 const int output_height = output_shape.Dims(1); in Mean()
194 const int output_width = output_shape.Dims(2); in Mean()
195 const int output_depth = output_shape.Dims(3); in Mean()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/
A Ddepthwiseconv_multithread.h39 const TS* bias_data, const RuntimeShape& output_shape, in DepthwiseConvWorkerTask()
49 output_shape_(output_shape), in DepthwiseConvWorkerTask()
79 inline int HowManyConvThreads(const RuntimeShape& output_shape, in HowManyConvThreads() argument
86 const int num_muls = output_shape.FlatSize() * filter_height * filter_width; in HowManyConvThreads()
123 const TS* bias_data, const RuntimeShape& output_shape, in DepthwiseConv() argument
130 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConv()
132 int thread_count = HowManyConvThreads(output_shape, filter_shape); in DepthwiseConv()
143 const int output_batches = output_shape.Dims(0); in DepthwiseConv()
144 const int output_height = output_shape.Dims(1); in DepthwiseConv()
151 filter_data, bias_shape, bias_data, output_shape, in DepthwiseConv()
[all …]

Completed in 4269 milliseconds

12345