Home
last modified time | relevance | path

Searched refs:FlatSizeSkipDim (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h41 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnected()
86 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
133 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
186 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected()
Dreference_ops.h491 TFLITE_DCHECK_EQ(FlatSizeSkipDim(bias_shape, 3), 1); in LstmCell()
704 TFLITE_DCHECK_EQ(FlatSizeSkipDim(bias_shape, 3), 1); in LstmCell()
710 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dconv.h84 const int gemm_input_cols = FlatSizeSkipDim(*gemm_input_shape, 3); in ConvPerChannel()
86 const int filter_cols = FlatSizeSkipDim(filter_shape, 0); in ConvPerChannel()
Dfully_connected.h54 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
Ddepthwise_conv_hybrid.h156 output_ptr_offset = batch_start * FlatSizeSkipDim(output_shape, 0); in DepthwiseConvHybridGeneral()
Ddepthwise_conv.h1727 output_ptr_offset = batch_start * FlatSizeSkipDim(output_shape, 0);
1953 FlatSizeSkipDim(output_shape, thread_dim) * filter_height * filter_width;
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h44 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnectedSparseWeight()
176 FlatSizeSkipDim(output_shape, output_shape.DimensionsCount() - 1); in FullyConnectedSparseWeight1x4()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h609 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) { in FlatSizeSkipDim() function
627 return FlatSizeSkipDim(dims, skip_dim); in MatchingFlatSizeSkipDim()
673 inline int FlatSizeSkipDim(const RuntimeShape& shape, int skip_dim) { in FlatSizeSkipDim() function
693 return FlatSizeSkipDim(shape, skip_dim); in MatchingFlatSizeSkipDim()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h758 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) {
776 return FlatSizeSkipDim(dims, skip_dim);
822 inline int FlatSizeSkipDim(const RuntimeShape& shape, int skip_dim) {
842 return FlatSizeSkipDim(shape, skip_dim);
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h431 FlatSizeSkipDim(output_shape, thread_dim) * filter_height * filter_width; in HowManyConvThreads()
784 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in LegacyFullyConnectedAsGEMVWorkerImpl()
785 const int input_size = FlatSizeSkipDim(input_shape, 0); in LegacyFullyConnectedAsGEMVWorkerImpl()
1053 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnectedAsGEMV()
1055 const int input_size = FlatSizeSkipDim(input_shape, 0); in FullyConnectedAsGEMV()
1113 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
1168 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCell()
1169 const int input_size = FlatSizeSkipDim(input_shape, 0); in GEMVForLstmCell()
1355 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCellWithSymmetricRange()
1356 const int input_size = FlatSizeSkipDim(input_shape, 0); in GEMVForLstmCellWithSymmetricRange()
[all …]
Doptimized_ops.h152 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsMatrixWithLastDimAsRows()
160 const int rows = FlatSizeSkipDim(shape, 0); in MapAsMatrixWithFirstDimAsCols()
176 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsArrayWithLastDimAsRows()
351 lhs_params.rows = FlatSizeSkipDim(weights_shape, dims_count - 1); in FullyConnected()
358 FlatSizeSkipDim(output_shape, output_shape.DimensionsCount() - 1); in FullyConnected()
391 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
457 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
835 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected()
1284 int m = FlatSizeSkipDim(*gemm_input_shape, gemm_input_dims - 1); in Conv()
1388 const int filter_cols = FlatSizeSkipDim(filter_shape, 0); in HybridConv()
[all …]
Ddepthwiseconv_float.h1016 output_ptr_offset = batch_start * FlatSizeSkipDim(output_shape, 0);
Ddepthwiseconv_uint8.h1794 output_ptr_offset = batch_start * FlatSizeSkipDim(output_shape, 0);
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected.cc602 FlatSizeSkipDim(output_shape, output_shape.DimensionsCount() - 1); in EvalHybrid()