Home
last modified time | relevance | path

Searched refs:FlatSize (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dlogsoftmax_quantized_test.cc42 const int ref_buffer_size = shape_common.FlatSize(); in RunLogSoftmaxFloatReference()
75 const int ref_buffer_size = shape_common.FlatSize(); in RunLogSoftmaxFloatReference()
104 const int buffer_size = shape_common.FlatSize(); in CheckOutputData()
149 const int buffer_size = shape_common.FlatSize(); in RunOneLogSoftmaxTest()
201 const int buffer_size = shape_common.FlatSize(); in RunOneLogSoftmaxTest()
260 const int buffer_size = shape_common.FlatSize(); in TryOneUniformLogSoftmax()
298 const int buffer_size = shape_common.FlatSize(); in TryOneSkyscraperLogSoftmax()
Dsoftmax_quantized_test.cc40 const int ref_buffer_size = shape_common.FlatSize(); in RunSoftmaxFloatReference()
67 const int buffer_size = shape_common.FlatSize(); in CheckOutputData()
104 const int buffer_size = shape_common.FlatSize(); in RunOneSoftmaxTest()
164 const int buffer_size = shape_common.FlatSize(); in TryOneUniformSoftmax()
202 const int buffer_size = shape_common.FlatSize(); in TryOneSkyscraperSoftmax()
Ddepthwiseconv_float_test.cc37 const int output_buffer_size = output_shape.FlatSize(); in TestOneDepthwiseConv()
115 const int input_buffer_size = input_shape_inference.FlatSize(); in TryTestOneDepthwiseConv()
116 const int filter_buffer_size = filter_shape_inference.FlatSize(); in TryTestOneDepthwiseConv()
Dresize_nearest_neighbor_test.cc161 std::vector<uint8> input_data(input_shape.FlatSize(), 0); in TestOptimizedResizeNearestNeighbor()
164 std::vector<uint8> reference_output_data(output_shape.FlatSize(), 0); in TestOptimizedResizeNearestNeighbor()
167 std::vector<uint8> output_data(output_shape.FlatSize(), 3); in TestOptimizedResizeNearestNeighbor()
Dresize_bilinear_test.cc35 const int input_buffer_size = input_dims_inference.FlatSize(); in TestOneResizeBilinear()
36 const int output_buffer_size = output_dims_inference.FlatSize(); in TestOneResizeBilinear()
Dtypes.h257 inline int FlatSize() const { in FlatSize() function
441 inline int FlatSize(const Dims<N>& dims) { in FlatSize() function
451 return FlatSize(dims); in RequiredBufferSizeForDims()
463 return shape.FlatSize(); in MatchingFlatSize()
509 return FlatSize(dims); in MatchingFlatSize()
Ddepthwiseconv_quantized_test.cc298 const int output_buffer_size = output_shape.FlatSize(); in TestOneDepthwiseConvWithGivenOutputShift()
533 const int input_buffer_size = input_shape_inference.FlatSize(); in TryTestDepthwiseConv()
534 const int filter_buffer_size = filter_shape_inference.FlatSize(); in TryTestDepthwiseConv()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h242 inline int FlatSize() const { in FlatSize() function
425 inline int FlatSize(const Dims<N>& dims) { in FlatSize() function
435 return FlatSize(dims); in RequiredBufferSizeForDims()
447 return shape.FlatSize(); in MatchingFlatSize()
493 return FlatSize(dims); in MatchingFlatSize()
/external/tensorflow/tensorflow/lite/kernels/
Dwhere.cc37 const int size = cond_shape.FlatSize(); in ResizeOutputTensor()
Dpad.cc49 const int paddings_total = GetTensorShape(paddings).FlatSize(); in PadContext()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h116 const int size = shape.FlatSize(); in MapAsVector()
181 const int flatsize = shape.FlatSize(); in MapAsMatrixWithGivenNumberOfRows()
195 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
196 const int array_size = array_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
246 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
247 const int array_size = array_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
807 const int flat_size = output_shape.FlatSize(); in FullyConnected()
1205 TFLITE_DCHECK_EQ(filter_shape.FlatSize(), filter_rows * filter_cols); in FullyConnected()
1208 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_rows); in FullyConnected()
2100 {1, 1, row_shape.FlatSize(), col_shape.FlatSize()}); in DilatedIm2col()
[all …]
Ddepthwiseconv_float.h923 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth);
Dlegacy_optimized_ops.h80 const int size = FlatSize(dims); in MapAsVector()
123 const int flatsize = FlatSize(dims); in MapAsMatrixWithGivenNumberOfRows()
Ddepthwiseconv_uint8.h1974 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth);
Ddepthwiseconv_uint8_3x3_filter.h7091 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth);
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_float.h54 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConv()
Ddepthwiseconv_uint8.h128 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Run()
Dreference_ops.h157 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Conv()
237 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Conv()
556 const size_t size = input_shape.FlatSize(); in AddN()
1729 TFLITE_DCHECK_EQ((**input_shapes).FlatSize(), copy_size * outer_size); in Pack()
1762 TFLITE_DCHECK_EQ(output_shape.FlatSize(), copy_size * outer_size); in Unpack()
1795 TFLITE_DCHECK_EQ((**input_shapes).FlatSize(), copy_size * outer_size); in PackWithScaling()
1890 TFLITE_DCHECK_EQ(weights_shape.FlatSize(), in LstmCell()
2096 TFLITE_DCHECK_EQ(weights_shape.FlatSize(), in LstmCell()
2948 const int coords_count = coords_shape.FlatSize(); in Gather()
2992 int remain_flat_size = params_shape.FlatSize(); in GatherNd()
[all …]
Dlegacy_reference_ops.h2042 TFLITE_DCHECK_EQ(output_shape.FlatSize(), in ArgMax()
2043 DimsToShape(output_dims).FlatSize()); in ArgMax()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dconv.h54 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in ConvPerChannel()
Ddepthwise_conv.h60 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvPerChannel()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h110 const int size = shape.FlatSize(); in MapAsVector()
116 const int size = FlatSize(dims); in MapAsVector()
216 const int flatsize = FlatSize(dims); in MapAsMatrixWithGivenNumberOfRows()
228 const int flatsize = shape.FlatSize(); in MapAsMatrixWithGivenNumberOfRows()
290 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
291 const int array_size = array_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
341 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
342 const int array_size = array_shape.FlatSize(); in AddBiasAndEvalActivationFunction()
1186 TFLITE_DCHECK_EQ(filter_shape.FlatSize(), filter_rows * filter_cols); in FullyConnected()
1189 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_rows); in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc128 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvOptimizedForFilterWidthEight()