Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h43 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnected()
88 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
135 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
187 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected()
Dreference_ops.h1172 TFLITE_DCHECK_EQ(FlatSizeSkipDim(bias_shape, 3), 1); in LstmCell()
1385 TFLITE_DCHECK_EQ(FlatSizeSkipDim(bias_shape, 3), 1); in LstmCell()
1391 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dconv.h85 const int gemm_input_cols = FlatSizeSkipDim(*gemm_input_shape, 3); in ConvPerChannel()
87 const int filter_cols = FlatSizeSkipDim(filter_shape, 0); in ConvPerChannel()
Dfully_connected.h52 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.h1719 output_ptr_offset = batch_start * FlatSizeSkipDim(output_shape, 0);
1910 FlatSizeSkipDim(output_shape, thread_dim) * filter_height * filter_width;
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h571 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) { in FlatSizeSkipDim() function
589 return FlatSizeSkipDim(dims, skip_dim); in MatchingFlatSizeSkipDim()
635 inline int FlatSizeSkipDim(const RuntimeShape& shape, int skip_dim) { in FlatSizeSkipDim() function
655 return FlatSizeSkipDim(shape, skip_dim); in MatchingFlatSizeSkipDim()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h432 FlatSizeSkipDim(output_shape, thread_dim) * filter_height * filter_width; in HowManyConvThreads()
783 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in LegacyFullyConnectedAsGEMVWorkerImpl()
784 const int input_size = FlatSizeSkipDim(input_shape, 0); in LegacyFullyConnectedAsGEMVWorkerImpl()
1052 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnectedAsGEMV()
1054 const int input_size = FlatSizeSkipDim(input_shape, 0); in FullyConnectedAsGEMV()
1112 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
1167 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCell()
1168 const int input_size = FlatSizeSkipDim(input_shape, 0); in GEMVForLstmCell()
1354 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCellWithSymmetricRange()
1355 const int input_size = FlatSizeSkipDim(input_shape, 0); in GEMVForLstmCellWithSymmetricRange()
[all …]
Doptimized_ops.h146 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsMatrixWithLastDimAsRows()
154 const int rows = FlatSizeSkipDim(shape, 0); in MapAsMatrixWithFirstDimAsCols()
170 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsArrayWithLastDimAsRows()
292 lhs_params.rows = FlatSizeSkipDim(weights_shape, dims_count - 1); in FullyConnected()
298 FlatSizeSkipDim(output_shape, output_shape.DimensionsCount() - 1); in FullyConnected()
331 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
395 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
771 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected()
1194 int m = FlatSizeSkipDim(*gemm_input_shape, gemm_input_dims - 1); in Conv()
1289 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);