Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h44 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnected()
90 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
138 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
192 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in ShuffledFullyConnected()
Dreference_ops.h1887 TFLITE_DCHECK_EQ(FlatSizeSkipDim(bias_shape, 3), 1); in LstmCell()
2098 TFLITE_DCHECK_EQ(FlatSizeSkipDim(bias_shape, 3), 1); in LstmCell()
2104 const int fc_batches = FlatSizeSkipDim(activ_temp_shape, 3); in LstmCell()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h530 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) { in FlatSizeSkipDim() function
548 return FlatSizeSkipDim(dims, skip_dim); in MatchingFlatSizeSkipDim()
594 inline int FlatSizeSkipDim(const RuntimeShape& shape, int skip_dim) { in FlatSizeSkipDim() function
614 return FlatSizeSkipDim(shape, skip_dim); in MatchingFlatSizeSkipDim()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h546 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) { in FlatSizeSkipDim() function
564 return FlatSizeSkipDim(dims, skip_dim); in MatchingFlatSizeSkipDim()
610 inline int FlatSizeSkipDim(const RuntimeShape& shape, int skip_dim) { in FlatSizeSkipDim() function
630 return FlatSizeSkipDim(shape, skip_dim); in MatchingFlatSizeSkipDim()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h135 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsMatrixWithLastDimAsRows()
143 const int rows = FlatSizeSkipDim(shape, 0); in MapAsMatrixWithFirstDimAsCols()
159 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsArrayWithLastDimAsRows()
307 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCell()
308 const int input_size = FlatSizeSkipDim(input_shape, 0); in GEMVForLstmCell()
494 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCellWithSymmetricRange()
495 const int input_size = FlatSizeSkipDim(input_shape, 0); in GEMVForLstmCellWithSymmetricRange()
829 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in FullyConnectedAsGEMVWorkerImpl()
830 const int input_size = FlatSizeSkipDim(input_shape, 0); in FullyConnectedAsGEMVWorkerImpl()
1101 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnectedAsGEMV()
[all …]
Dlegacy_optimized_ops.h755 const int input_cols = FlatSizeSkipDim(input_dims, 0); in ConvAsGemm()
757 const int filter_cols = FlatSizeSkipDim(filter_dims, 3); in ConvAsGemm()
759 const int output_cols = FlatSizeSkipDim(output_dims, 0); in ConvAsGemm()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h135 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsMatrixWithLastDimAsRows()
143 const int rows = FlatSizeSkipDim(shape, 0); in MapAsMatrixWithFirstDimAsCols()
192 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsArrayWithLastDimAsRows()
429 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCell()
430 const int input_size = FlatSizeSkipDim(input_shape, 0); in GEMVForLstmCell()
616 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCellWithSymmetricRange()
617 const int input_size = FlatSizeSkipDim(input_shape, 0); in GEMVForLstmCellWithSymmetricRange()
975 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in FullyConnectedAsGEMV()
976 const int input_size = FlatSizeSkipDim(input_shape, 0); in FullyConnectedAsGEMV()
1172 const int batches = FlatSizeSkipDim(output_shape, output_dim_count - 1); in FullyConnected()
[all …]