Home
last modified time | relevance | path

Searched refs:input2_dims (Results 1 – 12 of 12) sorted by relevance

/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h216 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()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h623 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv() argument
630 DimsToShape(input2_dims), input2_data, in BroadcastDiv()
636 const T* input2_data, const Dims<4>& input2_dims, in Div() argument
643 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Div()
980 const T* input2_data, const Dims<4>& input2_dims, in Comparison() argument
985 DimsToShape(input2_dims), input2_data, in Comparison()
993 const T* input2_data, const Dims<4>& input2_dims, in Comparison() argument
1009 DimsToShape(input2_dims), input2_data, in Comparison()
1017 const Dims<4>& input2_dims, bool* output_data, in BroadcastComparison() argument
1022 input1_data, DimsToShape(input2_dims), in BroadcastComparison()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dpack_test.cc56 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestPackTwoInputsFloat() local
64 CreateTensor(input2_data, input2_dims), in TestPackTwoInputsFloat()
88 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestPackThreeInputsFloat() local
97 CreateTensor(input2_data, input2_dims), in TestPackThreeInputsFloat()
123 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestPackTwoInputsQuantized() local
135 CreateQuantizedTensor(input2_data, input2_dims, 1.0, 128), in TestPackTwoInputsQuantized()
160 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestPackTwoInputsQuantized32() local
168 CreateTensor(input2_data, input2_dims), in TestPackTwoInputsQuantized32()
Dmaximum_minimum_test.cc33 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMaxMinFloat() local
42 CreateTensor(input2_data, input2_dims), in TestMaxMinFloat()
74 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMaxMinQuantized() local
84 CreateQuantizedTensor(input2_data, input2_dims, input2_scale, in TestMaxMinQuantized()
113 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMaxMinQuantizedInt32() local
122 CreateTensor(input2_data, input2_dims), in TestMaxMinQuantizedInt32()
Dconcatenation_test.cc35 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestConcatenateTwoInputs() local
42 CreateTensor(input2_data, input2_dims), in TestConcatenateTwoInputs()
78 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestConcatenateQuantizedTwoInputs() local
87 CreateQuantizedTensor(input2_data, input2_dims, input_scale, in TestConcatenateQuantizedTwoInputs()
Dmul_test.cc75 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMulFloat() local
84 CreateTensor(input2_data, input2_dims), in TestMulFloat()
102 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMulQuantized() local
112 CreateQuantizedTensor(input2_data, input2_quantized, input2_dims, in TestMulQuantized()
Dcomparisons_test.cc59 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonFloat() local
64 CreateTensor(input2_data, input2_dims), in TestComparisonFloat()
77 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonBool() local
82 CreateTensor(input2_data, input2_dims), in TestComparisonBool()
95 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonInt() local
100 CreateTensor(input2_data, input2_dims), in TestComparisonInt()
116 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonQuantizedUInt8() local
122 CreateQuantizedTensor(input2_data, input2_quantized, input2_dims, in TestComparisonQuantizedUInt8()
139 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonQuantizedInt8() local
145 CreateQuantizedTensor(input2_data, input2_quantized, input2_dims, in TestComparisonQuantizedInt8()
Dlogical_test.cc33 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestLogicalOp() local
42 CreateTensor(input2_data, input2_dims), in TestLogicalOp()
Dadd_test.cc95 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestAddFloat() local
103 CreateTensor(input2_data, input2_dims), in TestAddFloat()
122 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestAddQuantized() local
133 input2_dims, input2_scale, in TestAddQuantized()
Dsub_test.cc94 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestSubFloat() local
102 CreateTensor(input2_data, input2_dims), in TestSubFloat()
121 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestSubQuantized() local
132 input2_dims, input2_scale, in TestSubQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3437 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv() argument
3444 DimsToShape(input2_dims), input2_data, in BroadcastDiv()
3494 const float* input2_data, const Dims<4>& input2_dims, in Add() argument
3503 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
3511 const uint8* input2_data, const Dims<4>& input2_dims, in Add() argument
3542 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
3548 const int32* input2_data, const Dims<4>& input2_dims, in Add() argument
3557 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
3563 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd() argument
3570 DimsToShape(input2_dims), input2_data, in BroadcastAdd()
[all …]
/external/tensorflow/tensorflow/lite/micro/
Dmemory_helpers_test.cc198 const int input2_dims[] = {kDimsLen, 5, 5, 5, 5}; in TF_LITE_MICRO_TEST() local
203 nullptr, tflite::testing::IntArrayFromInts(input2_dims)); in TF_LITE_MICRO_TEST()