Searched refs:per_channel_shift (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util_test.cc | 234 std::vector<int> per_channel_shift(3); in TEST_F() local
242 per_channel_multiplier.data(), per_channel_shift.data())); in TEST_F()
245 EXPECT_THAT(per_channel_shift, ::testing::ElementsAre(-1, -2, -1)); in TEST_F()
343 std::vector<int> per_channel_shift(3); in TEST_F() local
351 per_channel_multiplier.data(), per_channel_shift.data())); in TEST_F()
354 EXPECT_THAT(per_channel_shift, ::testing::ElementsAre(-30, -31, 0)); in TEST_F()
|
| D | kernel_util.cc | 44 int32_t* per_channel_multiplier, int* per_channel_shift) { in PopulateConvolutionQuantizationParams() argument
84 per_channel_shift[i] = shift; in PopulateConvolutionQuantizationParams()
|
| D | kernel_util.h | 93 int32_t* per_channel_multiplier, int* per_channel_shift);
|