Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h42 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnected()
43 const int weights_dims_count = weights_shape.DimensionsCount(); in FullyConnected()
79 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
80 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
88 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected()
89 const int filter_dim_count = filter_shape.DimensionsCount(); in FullyConnected()
136 const int output_dim_count = output_shape.DimensionsCount(); in FullyConnected()
137 const int filter_dim_count = filter_shape.DimensionsCount(); in FullyConnected()
182 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in ShuffledFullyConnected()
183 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in ShuffledFullyConnected()
[all …]
Dreference_ops.h46 std::max(shape0.DimensionsCount(), shape1.DimensionsCount()); in ProcessBroadcastShapes()
147 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Conv()
148 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Conv()
149 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
230 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Conv()
231 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Conv()
232 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
295 TFLITE_DCHECK_LE(unextended_input_shape.DimensionsCount(), 4); in DepthToSpace()
296 TFLITE_DCHECK_LE(unextended_output_shape.DimensionsCount(), 4); in DepthToSpace()
348 TFLITE_DCHECK_LE(unextended_input_shape.DimensionsCount(), 4); in SpaceToDepth()
[all …]
Ddepthwiseconv_float.h40 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConv()
41 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConv()
42 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConv()
Ddepthwiseconv_uint8.h113 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Run()
114 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Run()
115 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Run()
Dsoftmax.h31 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax()
78 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h155 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) { in RuntimeShape()
173 inline int32 DimensionsCount() const { return size_; } in DimensionsCount() function
261 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount()); in RuntimeShape()
264 const int size_increase = new_shape_size - shape.DimensionsCount(); in RuntimeShape()
269 sizeof(int32) * shape.DimensionsCount()); in RuntimeShape()
282 const int dimensions_count = array_shape.DimensionsCount(); in ToRuntimeDims()
358 TFLITE_DCHECK_EQ(shape.DimensionsCount(), 4); in Offset()
442 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
443 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize()
453 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h158 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) { in RuntimeShape()
180 inline int32 DimensionsCount() const { return size_; } in DimensionsCount() function
278 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount()); in RuntimeShape()
280 const int size_increase = new_shape_size - shape.DimensionsCount(); in RuntimeShape()
285 sizeof(int32) * shape.DimensionsCount()); in RuntimeShape()
298 const int dimensions_count = array_shape.DimensionsCount(); in ToRuntimeDims()
374 TFLITE_DCHECK_EQ(shape.DimensionsCount(), 4); in Offset()
458 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
459 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize()
469 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
[all …]
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()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h37 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
38 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 2); in FullyConnected()
41 const int filter_dim_count = filter_shape.DimensionsCount(); in FullyConnected()
Dpooling.h29 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in AveragePool()
30 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
90 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in MaxPool()
91 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()
Ddepthwise_conv.h45 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
46 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
47 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
Dmean.h30 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
31 TFLITE_CHECK_LE(unextended_output_shape.DimensionsCount(), 4); in Mean()
Dsoftmax.h42 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h133 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows()
157 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows()
302 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in GEMVForLstmCell()
303 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in GEMVForLstmCell()
304 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in GEMVForLstmCell()
305 const int output_dim_count = output_shape.DimensionsCount(); in GEMVForLstmCell()
306 const int weights_dim_count = weights_shape.DimensionsCount(); in GEMVForLstmCell()
489 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in GEMVForLstmCellWithSymmetricRange()
490 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in GEMVForLstmCellWithSymmetricRange()
491 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in GEMVForLstmCellWithSymmetricRange()
[all …]
Dmultithreaded_conv.h149 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Conv()
150 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Conv()
151 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
Ddepthwiseconv_float.h909 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4);
910 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4);
911 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4);
Ddepthwiseconv_uint8.h1967 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4);
1968 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4);
1969 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4);
Ddepthwiseconv_uint8_3x3_filter.h7070 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4);
7071 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4);
7072 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4);
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h133 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows()
190 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows()
424 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in GEMVForLstmCell()
425 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in GEMVForLstmCell()
426 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in GEMVForLstmCell()
427 const int output_dim_count = output_shape.DimensionsCount(); in GEMVForLstmCell()
428 const int weights_dim_count = weights_shape.DimensionsCount(); in GEMVForLstmCell()
611 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in GEMVForLstmCellWithSymmetricRange()
612 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in GEMVForLstmCellWithSymmetricRange()
613 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in GEMVForLstmCellWithSymmetricRange()
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc113 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConvOptimizedForFilterWidthEight()
114 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConvOptimizedForFilterWidthEight()
115 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvOptimizedForFilterWidthEight()
/external/tensorflow/tensorflow/lite/kernels/
Dwhere.cc38 const int cond_rank = cond_shape.DimensionsCount(); in ResizeOutputTensor()
Dl2norm.cc110 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval()
Dactivations.cc866 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmaxEval()