• Home
  • Raw
  • Download

Lines Matching refs:input2_dims

216     std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims;  in TestQS8()  local
219 std::fill(input2_dims.begin(), input2_dims.end(), 1); in TestQS8()
222 … std::copy(input2_shape().cbegin(), input2_shape().cend(), input2_dims.end() - num_input2_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()
239 input2_strides[i - 1] = input2_dims[i - 1] == 1 ? 0 : input2_stride; in TestQS8()
242 input2_stride *= input2_dims[i - 1]; in TestQS8()
354 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestF16() local
357 std::fill(input2_dims.begin(), input2_dims.end(), 1); in TestF16()
360 … std::copy(input2_shape().cbegin(), input2_shape().cend(), input2_dims.end() - num_input2_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()
377 input2_strides[i - 1] = input2_dims[i - 1] == 1 ? 0 : input2_stride; in TestF16()
380 input2_stride *= input2_dims[i - 1]; in TestF16()
505 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestF32() local
508 std::fill(input2_dims.begin(), input2_dims.end(), 1); in TestF32()
511 … std::copy(input2_shape().cbegin(), input2_shape().cend(), input2_dims.end() - num_input2_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()
528 input2_strides[i - 1] = input2_dims[i - 1] == 1 ? 0 : input2_stride; in TestF32()
531 input2_stride *= input2_dims[i - 1]; in TestF32()