• Home
  • Raw
  • Download

Lines Matching refs:input1_dims

215     std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims;  in TestQS8()  local
218 std::fill(input1_dims.begin(), input1_dims.end(), 1); in TestQS8()
221 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestQS8()
224 if (input1_dims[i] != 1 && input2_dims[i] != 1) { in TestQS8()
225 ASSERT_EQ(input1_dims[i], input2_dims[i]); in TestQS8()
227 output_dims[i] = std::max(input1_dims[i], input2_dims[i]); in TestQS8()
238 input1_strides[i - 1] = input1_dims[i - 1] == 1 ? 0 : input1_stride; in TestQS8()
241 input1_stride *= input1_dims[i - 1]; in TestQS8()
353 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestF16() local
356 std::fill(input1_dims.begin(), input1_dims.end(), 1); in TestF16()
359 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestF16()
362 if (input1_dims[i] != 1 && input2_dims[i] != 1) { in TestF16()
363 ASSERT_EQ(input1_dims[i], input2_dims[i]); in TestF16()
365 output_dims[i] = std::max(input1_dims[i], input2_dims[i]); in TestF16()
376 input1_strides[i - 1] = input1_dims[i - 1] == 1 ? 0 : input1_stride; in TestF16()
379 input1_stride *= input1_dims[i - 1]; in TestF16()
504 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestF32() local
507 std::fill(input1_dims.begin(), input1_dims.end(), 1); in TestF32()
510 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestF32()
513 if (input1_dims[i] != 1 && input2_dims[i] != 1) { in TestF32()
514 ASSERT_EQ(input1_dims[i], input2_dims[i]); in TestF32()
516 output_dims[i] = std::max(input1_dims[i], input2_dims[i]); in TestF32()
527 input1_strides[i - 1] = input1_dims[i - 1] == 1 ? 0 : input1_stride; in TestF32()
530 input1_stride *= input1_dims[i - 1]; in TestF32()