/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | fully_connected.h | 42 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 …]
|
D | reference_ops.h | 46 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 …]
|
D | depthwiseconv_float.h | 40 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()
|
D | depthwiseconv_uint8.h | 113 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()
|
D | softmax.h | 31 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/ |
D | types.h | 155 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/ |
D | types.h | 158 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 …]
|
D | tensor_test.cc | 27 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/ |
D | fully_connected.h | 37 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()
|
D | pooling.h | 29 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()
|
D | conv.h | 47 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()
|
D | depthwise_conv.h | 45 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()
|
D | mean.h | 30 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean() 31 TFLITE_CHECK_LE(unextended_output_shape.DimensionsCount(), 4); in Mean()
|
D | softmax.h | 42 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax()
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 133 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 …]
|
D | multithreaded_conv.h | 149 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()
|
D | depthwiseconv_float.h | 909 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); 910 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); 911 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4);
|
D | depthwiseconv_uint8.h | 1967 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); 1968 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); 1969 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4);
|
D | depthwiseconv_uint8_3x3_filter.h | 7070 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/ |
D | optimized_ops.h | 133 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/ |
D | depthwise_conv.cc | 113 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/ |
D | where.cc | 38 const int cond_rank = cond_shape.DimensionsCount(); in ResizeOutputTensor()
|
D | l2norm.cc | 110 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval()
|
D | activations.cc | 866 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmaxEval()
|