Home
last modified time | relevance | path

Searched refs:MatchingDim (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpooling.h32 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool()
33 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in AveragePool()
88 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool()
89 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in AveragePool()
141 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in L2Pool()
142 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in L2Pool()
195 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool()
196 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in MaxPool()
249 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool()
250 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in MaxPool()
Dconv.h45 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv()
46 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv()
47 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv()
128 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv()
129 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv()
130 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv()
208 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in HybridConvPerChannel()
209 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in HybridConvPerChannel()
210 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in HybridConvPerChannel()
Dtranspose_conv.h41 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConv()
42 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv()
43 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv()
131 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConv()
132 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv()
133 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv()
Dconv3d.h33 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv3D()
34 const int input_num_channels = MatchingDim(input_shape, 4, filter_shape, 3); in Conv3D()
35 const int output_num_channels = MatchingDim(filter_shape, 4, output_shape, 4); in Conv3D()
Dl2normalization.h38 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim);
60 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
Ddepthwiseconv_uint8.h117 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Run()
118 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in Run()
204 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in RunPerChannel()
205 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in RunPerChannel()
Dresize_nearest_neighbor.h62 int32_t batches = MatchingDim(input_shape, 0, output_shape, 0); in ResizeNearestNeighbor()
65 int32_t depth = MatchingDim(input_shape, 3, output_shape, 3); in ResizeNearestNeighbor()
Ddepthwiseconv_float.h44 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()
Dfully_connected.h42 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnected()
87 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected()
134 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected()
187 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected()
Dsoftmax.h37 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
90 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
174 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in SoftmaxInt16()
Dconcatenation.h43 MatchingDim(*input_shapes[i], j, output_shape, j); in Concatenation()
94 MatchingDim(*input_shapes[i], j, output_shape, j); in ConcatenationWithScaling()
Dreference_ops.h478 MatchingDim(input_shape, 0, prev_activ_shape, 0, prev_state_shape, 0, in LstmCell()
481 MatchingDim(input_shape, 1, prev_activ_shape, 1, prev_state_shape, 1, in LstmCell()
484 MatchingDim(input_shape, 2, prev_activ_shape, 2, prev_state_shape, 2, in LstmCell()
493 MatchingDim(weights_shape, weights_dim_count - 2, bias_shape, 3); in LstmCell()
498 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
701 MatchingDim(weights_shape, weights_dim_count - 2, bias_shape, 3); in LstmCell()
707 MatchingDim(prev_state_shape, 3, prev_activ_shape, 3, output_state_shape, in LstmCell()
712 MatchingDim(weights_shape, weights_dim_count - 2, activ_temp_shape, 3); in LstmCell()
847 MatchingDim(*output_shapes[i], j, input_shape, j); in Split()
887 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in LocalResponseNormalization()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dpooling.h32 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool()
33 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in AveragePool()
93 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool()
94 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in MaxPool()
148 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool()
149 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in AveragePool()
209 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool()
210 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in MaxPool()
Dconv.h50 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()
153 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in ConvPerChannel()
154 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel()
155 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel()
Dtranspose_conv.h42 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConv()
43 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv()
44 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv()
141 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConv()
142 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv()
143 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv()
Ddepthwise_conv.h49 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvPerChannel()
50 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel()
146 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvPerChannel()
147 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel()
229 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DepthwiseConvHybridPerChannel()
230 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvHybridPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dtranspose_conv.h38 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConvV2()
44 MatchingDim(input_shape, 3, hwoi_ordered_filter_shape, 3); in TransposeConvV2()
46 MatchingDim(output_shape, 3, hwoi_ordered_filter_shape, 2); in TransposeConvV2()
Dpooling.h53 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxPool()
54 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in MaxPool()
163 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in AveragePool()
164 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in AveragePool()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h138 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in DilatedIm2col()
141 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in DilatedIm2col()
146 MatchingDim(output_shape, 3, filter_shape, 0); in DilatedIm2col()
224 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Im2col()
260 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Im2col()
Dmultithreaded_conv.h160 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in Conv()
161 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv()
162 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv()
/external/tensorflow/tensorflow/lite/kernels/perception/
Ddense_image_warp.cc38 const int batches = MatchingDim(input_shape, 0, flow_shape, 0); in DenseImageWarp()
39 const int height = MatchingDim(input_shape, 1, flow_shape, 1); in DenseImageWarp()
40 const int width = MatchingDim(input_shape, 2, flow_shape, 2); in DenseImageWarp()
Dmax_unpooling_2d.cc37 const int batches = MatchingDim(input_shape, 0, output_shape, 0); in MaxUnpooling()
38 const int depth = MatchingDim(input_shape, 3, output_shape, 3); in MaxUnpooling()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h45 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnectedSparseWeight()
92 const int input_depth = MatchingDim(weights_shape, weights_dims_count - 1, in FullyConnectedSparseWeight1x4Impl()
94 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnectedSparseWeight1x4Impl()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dconv.cc64 input_dims.n = MatchingDim(input_shape, 0, output_shape, 0); in Prepare()
167 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in EvalQuantizedPerChannel()
168 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in EvalQuantizedPerChannel()
169 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in EvalQuantizedPerChannel()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsoftmax.cc72 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in SoftmaxHifimini()
196 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in PrepareHifi4()
222 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in EvalHifi4()

12