Lines Matching refs:input_stride
37 inline SoftMaxOperatorTester& input_stride(size_t input_stride) { in input_stride() function
38 assert(input_stride != 0); in input_stride()
39 this->input_stride_ = input_stride; in input_stride()
43 inline size_t input_stride() const { in input_stride() function
119 std::vector<uint8_t> input((batch_size() - 1) * input_stride() + channels()); in TestQ8()
129 input.data() + i * input_stride(), in TestQ8()
130 input.data() + i * input_stride() + channels()); in TestQ8()
134 std::exp((int32_t(input[i * input_stride() + c]) - max_input) * in TestQ8()
139 std::exp((int32_t(input[i * input_stride() + c]) - max_input) * in TestQ8()
152 channels(), input_stride(), output_stride(), in TestQ8()
185 …std::vector<float> input((batch_size() - 1) * input_stride() + channels() + XNN_EXTRA_BYTES / size… in TestF32()
195 input.data() + i * input_stride(), in TestF32()
196 input.data() + i * input_stride() + channels()); in TestF32()
199 sum_exp += std::exp(double(input[i * input_stride() + c]) - max_input); in TestF32()
203 std::exp(double(input[i * input_stride() + c]) - max_input) / sum_exp; in TestF32()
213 channels(), input_stride(), output_stride(), in TestF32()