Home
last modified time | relevance | path

Searched refs:output_shift_per_channel (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc407 params.output_shift_per_channel, input_shape, input_data, in DispatchDepthwiseConvImpl()
433 params.output_shift_per_channel); in DispatchDepthwiseConvImpl()
458 params.output_shift_per_channel); in DispatchDepthwiseConvImpl()
481 EXPECT_NE(params.output_shift_per_channel, nullptr); in DispatchDepthwiseConvImpl()
484 bias_shape, bias_data, params.output_shift_per_channel, in DispatchDepthwiseConvImpl()
513 EXPECT_NE(params.output_shift_per_channel, nullptr); in DispatchDepthwiseConvImpl()
519 params.output_shift_per_channel, input_shape, input_data, filter_shape, in DispatchDepthwiseConvImpl()
595 op_params.output_shift_per_channel, input_shape, input_data, in Run()
650 std::vector<int32> output_shift_per_channel(depth, -output_shift); in TestOneDepthwiseConvWithGivenOutputShift() local
654 output_shift_per_channel[i] += output_shift_adjust[i]; in TestOneDepthwiseConvWithGivenOutputShift()
[all …]
Dtypes.h914 const int32_t* output_shift_per_channel; member
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h5610 const int32* output_shift_per_channel =
5611 function_params->output_shift_per_channel;
5624 TFLITE_DCHECK_NE(output_shift_per_channel, nullptr);
5684 vld1q_s32(output_shift_per_channel + j_depth * 8 + 4 * s);
6106 vld1q_s32(output_shift_per_channel + j_depth * 8);
6110 vld1q_s32(output_shift_per_channel + j_depth * 8 + 4);
6282 const int32* output_shift_per_channel =
6283 function_params->output_shift_per_channel;
6296 TFLITE_DCHECK_NE(output_shift_per_channel, nullptr);
6338 vld1q_s32(output_shift_per_channel + j_depth * 8 + 4 * s);
[all …]
Ddepthwiseconv_3x3_filter_common.h374 const int32* output_shift_per_channel;
Ddepthwiseconv_uint8_3x3_filter.h310 static_assert(offsetof(DepthwiseConvDotProdParams, output_shift_per_channel) ==
13305 function_params.output_shift_per_channel =
13306 params.output_shift_per_channel + 64 * j_depth;
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_uint8.h196 const int32_t* output_shift = params.output_shift_per_channel; in RunPerChannel()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h1013 const int32_t* output_shift_per_channel;
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv.h1835 params_copy.output_shift_per_channel = output_shift;