| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | pooling.h | 31 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool()
32 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in AveragePool()
92 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool()
93 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in MaxPool()
|
| D | conv.h | 50 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in ConvPerChannel()
51 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel()
52 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel()
|
| D | depthwise_conv.h | 50 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvPerChannel()
51 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel()
|
| D | softmax.h | 46 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 153 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv()
154 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv()
155 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv()
|
| D | optimized_ops.h | 309 const int output_size = MatchingDim(weights_shape, weights_dim_count - 2, in GEMVForLstmCell()
496 const int output_size = MatchingDim(weights_shape, weights_dim_count - 2, in GEMVForLstmCellWithSymmetricRange()
1192 const int output_size = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected()
1255 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected()
1657 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected()
2083 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DilatedIm2col()
2086 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in DilatedIm2col()
2091 MatchingDim(output_shape, 3, filter_shape, 0); in DilatedIm2col()
2157 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Im2col()
2570 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
[all …]
|
| D | depthwiseconv_float.h | 913 const int batches = MatchingDim(input_shape, 0, output_shape, 0);
914 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3);
|
| D | depthwiseconv_uint8.h | 1680 const int batches = MatchingDim(input_shape, 0, output_shape, 0);
1681 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3);
1971 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3);
|
| D | depthwiseconv_uint8_3x3_filter.h | 3563 params.output_depth = MatchingDim(filter_shape, 3, output_shape, 3);
3591 const int32 batches = MatchingDim(input_shape, 0, output_shape, 0);
7083 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3);
7088 const int batches = MatchingDim(input_shape, 0, output_shape, 0);
7093 TFLITE_DCHECK_EQ(MatchingDim(filter_shape, 1, filter_shape, 2), filter_size);
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | fully_connected.h | 45 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnected()
91 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected()
139 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected()
193 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected()
|
| D | depthwiseconv_float.h | 44 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConv()
45 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConv()
|
| D | reference_ops.h | 153 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv()
154 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv()
155 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv()
233 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv()
234 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv()
235 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv()
478 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
499 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
1617 MatchingDim(*input_shapes[i], j, output_shape, j); in Concatenation()
1670 MatchingDim(*input_shapes[i], j, output_shape, j); in ConcatenationWithScaling()
[all …]
|
| D | softmax.h | 35 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
82 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
|
| D | depthwiseconv_uint8.h | 118 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Run()
119 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in Run()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 431 const int output_size = MatchingDim(weights_shape, weights_dim_count - 2, in GEMVForLstmCell()
618 const int output_size = MatchingDim(weights_shape, weights_dim_count - 2, in GEMVForLstmCellWithSymmetricRange()
977 const int output_size = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnectedAsGEMV()
1174 const int output_size = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected()
1263 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected()
1713 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected()
1960 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DilatedIm2col()
1963 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in DilatedIm2col()
1968 MatchingDim(output_shape, 3, filter_shape, 0); in DilatedIm2col()
2058 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Im2col()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | l2norm.cc | 112 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Eval()
|
| D | activations.cc | 870 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in LogSoftmaxEval()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/ |
| D | depthwise_conv.cc | 118 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvOptimizedForFilterWidthEight()
119 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvOptimizedForFilterWidthEight()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 426 inline int MatchingDim(const RuntimeShape& shape1, int index1, in MatchingDim() function
433 int MatchingDim(const RuntimeShape& shape1, int index1, in MatchingDim() function
436 return MatchingDim(shape1, index1, args...); in MatchingDim()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 410 inline int MatchingDim(const RuntimeShape& shape1, int index1, in MatchingDim() function
417 int MatchingDim(const RuntimeShape& shape1, int index1, in MatchingDim() function
420 return MatchingDim(shape1, index1, args...); in MatchingDim()
|