Home
last modified time | relevance | path

Searched refs:DimensionsCount (Results 1 – 25 of 86) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbatch_to_space_nd.h28 if (shape.DimensionsCount() == 4) { in ExtendShapeBatchToSpace()
48 TFLITE_DCHECK_GE(unextended_input1_shape.DimensionsCount(), 3); in BatchToSpaceND()
49 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in BatchToSpaceND()
50 TFLITE_DCHECK_EQ(unextended_input1_shape.DimensionsCount(), in BatchToSpaceND()
51 unextended_output_shape.DimensionsCount()); in BatchToSpaceND()
69 unextended_input1_shape.DimensionsCount() == 4 ? block_shape_data[1] : 1; in BatchToSpaceND()
72 unextended_input1_shape.DimensionsCount() == 4 ? crops_data[2] : 0; in BatchToSpaceND()
Dspace_to_batch_nd.h29 if (shape.DimensionsCount() == 4) { in ExtendShapeSpaceToBatch()
50 TFLITE_DCHECK_GE(unextended_input1_shape.DimensionsCount(), 3); in SpaceToBatchND()
51 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), 4); in SpaceToBatchND()
52 TFLITE_DCHECK_EQ(unextended_input1_shape.DimensionsCount(), in SpaceToBatchND()
53 unextended_output_shape.DimensionsCount()); in SpaceToBatchND()
72 unextended_input1_shape.DimensionsCount() == 4 ? block_shape_data[1] : 1; in SpaceToBatchND()
75 unextended_input1_shape.DimensionsCount() == 4 ? paddings_data[2] : 0; in SpaceToBatchND()
Dfully_connected.h39 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnected()
40 const int weights_dims_count = weights_shape.DimensionsCount(); in FullyConnected()
75 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
76 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
84 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected()
85 const int filter_dim_count = filter_shape.DimensionsCount(); in FullyConnected()
131 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected()
132 const int filter_dim_count = filter_shape.DimensionsCount(); in FullyConnected()
176 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in ShuffledFullyConnected()
177 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in ShuffledFullyConnected()
[all …]
Darg_min_max.h28 TFLITE_DCHECK_GT(input1_shape.DimensionsCount(), 0); in ArgMinMax()
29 TFLITE_DCHECK_EQ(input1_shape.DimensionsCount() - 1, in ArgMinMax()
30 output_shape.DimensionsCount()); in ArgMinMax()
33 axis += input1_shape.DimensionsCount(); in ArgMinMax()
44 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax()
Dreference_ops.h321 const int dimensions = output_shape.DimensionsCount(); in Pack()
349 const int dimensions = input_shape.DimensionsCount(); in Unpack()
383 const int dimensions = output_shape.DimensionsCount(); in PackWithScaling()
450 TFLITE_DCHECK_LE(unextended_input_shape.DimensionsCount(), 4); in LstmCell()
451 TFLITE_DCHECK_LE(unextended_prev_activ_shape.DimensionsCount(), 4); in LstmCell()
452 TFLITE_DCHECK_LE(unextended_bias_shape.DimensionsCount(), 4); in LstmCell()
453 TFLITE_DCHECK_LE(unextended_prev_state_shape.DimensionsCount(), 4); in LstmCell()
454 TFLITE_DCHECK_LE(unextended_output_state_shape.DimensionsCount(), 4); in LstmCell()
455 TFLITE_DCHECK_LE(unextended_output_activ_shape.DimensionsCount(), 4); in LstmCell()
456 TFLITE_DCHECK_LE(unextended_concat_temp_shape.DimensionsCount(), 4); in LstmCell()
[all …]
Dpooling.h30 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in AveragePool()
31 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
86 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in AveragePool()
87 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
139 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in L2Pool()
140 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in L2Pool()
193 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in MaxPool()
194 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in MaxPool()
247 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in MaxPool()
248 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in MaxPool()
Dconv.h39 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Conv()
40 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Conv()
41 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
125 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Conv()
126 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Conv()
127 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
205 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in HybridConvPerChannel()
206 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in HybridConvPerChannel()
207 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in HybridConvPerChannel()
Dsub.h77 TFLITE_DCHECK_LE(input1_shape.DimensionsCount(), N); in BroadcastSubSlow()
78 TFLITE_DCHECK_LE(input2_shape.DimensionsCount(), N); in BroadcastSubSlow()
79 TFLITE_DCHECK_LE(output_shape.DimensionsCount(), N); in BroadcastSubSlow()
117 TFLITE_DCHECK_LE(input1_shape.DimensionsCount(), N); in BroadcastSubSlow()
118 TFLITE_DCHECK_LE(input2_shape.DimensionsCount(), N); in BroadcastSubSlow()
119 TFLITE_DCHECK_LE(output_shape.DimensionsCount(), N); in BroadcastSubSlow()
174 TFLITE_DCHECK_LE(input1_shape.DimensionsCount(), N); in BroadcastSubSlow()
175 TFLITE_DCHECK_LE(input2_shape.DimensionsCount(), N); in BroadcastSubSlow()
176 TFLITE_DCHECK_LE(output_shape.DimensionsCount(), N); in BroadcastSubSlow()
267 TFLITE_DCHECK_LE(input1_shape.DimensionsCount(), N); in BroadcastSubSlow()
[all …]
Ddiv.h81 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), N); in BroadcastDivSlow()
82 TFLITE_DCHECK_LE(unextended_input2_shape.DimensionsCount(), N); in BroadcastDivSlow()
83 TFLITE_DCHECK_LE(unextended_output_shape.DimensionsCount(), N); in BroadcastDivSlow()
143 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), N); in BroadcastDivSlow()
144 TFLITE_DCHECK_LE(unextended_input2_shape.DimensionsCount(), N); in BroadcastDivSlow()
145 TFLITE_DCHECK_LE(unextended_output_shape.DimensionsCount(), N); in BroadcastDivSlow()
Dmaximum_minimum.h40 TFLITE_DCHECK_LE(unextended_input1_shape.DimensionsCount(), N); in MaximumMinimumBroadcastSlow()
41 TFLITE_DCHECK_LE(unextended_input2_shape.DimensionsCount(), N); in MaximumMinimumBroadcastSlow()
42 TFLITE_DCHECK_LE(unextended_output_shape.DimensionsCount(), N); in MaximumMinimumBroadcastSlow()
Dgather.h33 axis += input_shape.DimensionsCount(); in Gather()
36 TFLITE_DCHECK_LT(axis, input_shape.DimensionsCount()); in Gather()
46 for (int i = axis + 1; i < input_shape.DimensionsCount(); ++i) { in Gather()
Dtranspose_conv.h35 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in TransposeConv()
36 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in TransposeConv()
37 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in TransposeConv()
125 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in TransposeConv()
126 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in TransposeConv()
127 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in TransposeConv()
Dconcatenation.h35 const int concat_dimensions = output_shape.DimensionsCount(); in Concatenation()
40 TFLITE_DCHECK_EQ(input_shapes[i]->DimensionsCount(), concat_dimensions); in Concatenation()
86 const int concat_dimensions = output_shape.DimensionsCount(); in ConcatenationWithScaling()
91 TFLITE_DCHECK_EQ(input_shapes[i]->DimensionsCount(), concat_dimensions); in ConcatenationWithScaling()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_test.cc27 std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()), in TEST()
34 std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()), in TEST()
41 std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()), in TEST()
48 std::vector<int32>(d.DimsData(), d.DimsData() + d.DimensionsCount()), in TEST()
Dtypes.h185 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) { in RuntimeShape()
212 inline int32_t DimensionsCount() const { return size_; } in DimensionsCount() function
308 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount()); in RuntimeShape()
310 const int size_increase = new_shape_size - shape.DimensionsCount(); in RuntimeShape()
315 sizeof(int32_t) * shape.DimensionsCount()); in RuntimeShape()
328 const int dimensions_count = array_shape.DimensionsCount(); in ToRuntimeDims()
404 TFLITE_DCHECK_EQ(shape.DimensionsCount(), 4); in Offset()
415 TFLITE_DCHECK_EQ(shape.DimensionsCount(), 5); in Offset()
521 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
522 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h36 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
37 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 2); in FullyConnected()
40 const int filter_dim_count = filter_shape.DimensionsCount(); in FullyConnected()
76 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
77 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 2); in FullyConnected()
80 const int filter_dim_count = filter_shape.DimensionsCount(); in FullyConnected()
Ddepthwise_conv.h44 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
45 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
46 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
141 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
142 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
143 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
225 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConvHybridPerChannel()
226 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConvHybridPerChannel()
227 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvHybridPerChannel()
Dpooling.h30 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in AveragePool()
31 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
91 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in MaxPool()
92 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in MaxPool()
146 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in AveragePool()
147 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
207 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in MaxPool()
208 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in MaxPool()
Dconv.h47 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in ConvPerChannel()
48 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in ConvPerChannel()
49 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in ConvPerChannel()
150 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in ConvPerChannel()
151 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in ConvPerChannel()
152 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in ConvPerChannel()
Dtranspose_conv.h36 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in TransposeConv()
37 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in TransposeConv()
38 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in TransposeConv()
135 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in TransposeConv()
136 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in TransposeConv()
137 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in TransposeConv()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h353 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) {
375 inline int32_t DimensionsCount() const { return size_; }
471 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount());
473 const int size_increase = new_shape_size - shape.DimensionsCount();
478 sizeof(int32_t) * shape.DimensionsCount());
491 const int dimensions_count = array_shape.DimensionsCount();
567 TFLITE_DCHECK_EQ(shape.DimensionsCount(), 4);
670 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount());
671 const int dims_count = shape.DimensionsCount();
681 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount());
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h45 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
46 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
52 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected()
53 const int filter_dim_count = filter_shape.DimensionsCount(); in FullyConnected()
/external/tensorflow/tensorflow/lite/kernels/
Dscatter_nd.cc54 TF_LITE_ENSURE(context, (indices.DimensionsCount() >= 1) && in CheckShapes()
55 (updates.DimensionsCount() >= 1) && in CheckShapes()
56 (shape_shape.DimensionsCount() == 1)); in CheckShapes()
58 const int outer_dims = indices.DimensionsCount() - 1; in CheckShapes()
64 TF_LITE_ENSURE_EQ(context, updates.DimensionsCount() - outer_dims, in CheckShapes()
66 for (int i = 0; i + outer_dims < updates.DimensionsCount(); ++i) { in CheckShapes()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h36 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in ExtractPatchIntoBufferColumn()
128 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DilatedIm2col()
129 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DilatedIm2col()
130 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DilatedIm2col()
221 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Im2col()
222 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Im2col()
257 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Im2col()
258 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Im2col()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h42 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnectedSparseWeight()
43 const int weights_dims_count = weights_shape.DimensionsCount(); in FullyConnectedSparseWeight()
88 const int input_dims_count = input_shape.DimensionsCount(); in FullyConnectedSparseWeight1x4Impl()
89 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnectedSparseWeight1x4Impl()
90 const int weights_dims_count = weights_shape.DimensionsCount(); in FullyConnectedSparseWeight1x4Impl()
176 FlatSizeSkipDim(output_shape, output_shape.DimensionsCount() - 1); in FullyConnectedSparseWeight1x4()

1234