Home
last modified time | relevance | path

Searched refs:output_rounding (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid_3x3_filter.h117 template <DepthwiseConvOutputRounding output_rounding, int32 kDepth,
121 template <DepthwiseConvOutputRounding output_rounding, EdgeType kEdgeType,
2785 template <DepthwiseConvOutputRounding output_rounding, int32 kStrideWidth,
2799 DepthwiseConvHybridWindowPerChannel<output_rounding, 8, kStrideWidth,
2818 template <DepthwiseConvOutputRounding output_rounding, int32 kStrideWidth,
2822 DepthwiseConvHybridThroughDepthPerChannel<output_rounding, kStrideWidth,
2932 template <DepthwiseConvOutputRounding output_rounding>
2940 DepthwiseConvHybridPartialPerChannel<output_rounding, EdgeType::kCenter, 1,
2959 output_rounding, EdgeType::kCorner, 1, 1>::Run(
2970 output_rounding, EdgeType::kHorizontal, 1, 1>::Run(
[all …]
Ddepthwise_conv_3x3_filter.h2617 template <DepthwiseConvOutputRounding output_rounding, int32 kStrideWidth,
2631 DepthwiseConvWindowPerChannel<output_rounding, 8, kStrideWidth,
2651 template <DepthwiseConvOutputRounding output_rounding, int32 kStrideWidth,
2655 DepthwiseConvThroughDepthPerChannel<output_rounding, kStrideWidth,
2764 template <DepthwiseConvOutputRounding output_rounding>
2772 DepthwiseConvPartialPerChannel<output_rounding, EdgeType::kCenter, 1,
2791 DepthwiseConvPartialPerChannel<output_rounding, EdgeType::kCorner, 1, 1>::Run(
2801 DepthwiseConvPartialPerChannel<output_rounding, EdgeType::kHorizontal, 1,
2810 DepthwiseConvPartialPerChannel<output_rounding, EdgeType::kCorner, 1, 1>::Run(
2821 DepthwiseConvPartialPerChannel<output_rounding, EdgeType::kVertical, 1,
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_uint8.h64 template <DepthwiseConvOutputRounding output_rounding>
67 TFLITE_DCHECK_NE(output_rounding, DepthwiseConvOutputRounding::kNone); in DepthwiseConvRound()
90 template <DepthwiseConvOutputRounding output_rounding>
160 acc = DepthwiseConvRound<output_rounding>(acc, output_multiplier, in Run()
262 acc = DepthwiseConvRound<output_rounding>( in RunPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc92 output_rounding(::testing::get<6>(param_tuple)), in TestParam()
113 DepthwiseConvOutputRounding output_rounding = member
196 switch (test_param.output_rounding) { in DispatchDepthwiseConvImpl()
402 switch (test_param.output_rounding) { in DispatchDepthwiseConvImpl()
553 switch (test_param.output_rounding) { in Run()
569 EXPECT_NE(test_param.output_rounding, in Run()
584 switch (test_param.output_rounding) { in Run()
601 EXPECT_NE(test_param.output_rounding, in Run()
1055 DepthwiseConvOutputRounding output_rounding) { in TestOneDepthwiseConv() argument
1058 test_param.output_rounding = output_rounding; in TestOneDepthwiseConv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_3x3_filter_common.h181 template <DepthwiseConvOutputRounding output_rounding>
377 template <DepthwiseConvOutputRounding output_rounding, int32 kDepth,
381 template <DepthwiseConvOutputRounding output_rounding, int32 kDepth,
387 template <DepthwiseConvOutputRounding output_rounding, EdgeType kEdgeType,
391 template <DepthwiseConvOutputRounding output_rounding, EdgeType kEdgeType,
Ddepthwiseconv_uint8_3x3_filter.h5362 template <DepthwiseConvOutputRounding output_rounding, int32 kStrideWidth,
5375 DepthwiseConvWindow<output_rounding, 8, kStrideWidth, kStrideHeight>::Run(
5386 template <DepthwiseConvOutputRounding output_rounding, int32 kStrideWidth,
5390 DepthwiseConvThroughDepth<output_rounding, kStrideWidth, kStrideHeight>;
5489 template <DepthwiseConvOutputRounding output_rounding>
5498 DepthwiseConvPartial<output_rounding, EdgeType::kCenter, 1, 1>::Run(
5514 DepthwiseConvPartial<output_rounding, EdgeType::kCorner, 1, 1>::Run(
5523 DepthwiseConvPartial<output_rounding, EdgeType::kHorizontal, 1, 1>::Run(
5529 DepthwiseConvPartial<output_rounding, EdgeType::kCorner, 1, 1>::Run(
5539 DepthwiseConvPartial<output_rounding, EdgeType::kVertical, 1, 1>::Run(
[all …]