Home
last modified time | relevance | path

Searched refs:input1_dims (Results 1 – 15 of 15) sorted by relevance

/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h215 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()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h622 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv() argument
629 BroadcastDivSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastDiv()
635 inline void Div(const T* input1_data, const Dims<4>& input1_dims, in Div() argument
642 Div(op_params, DimsToShape(input1_dims), input1_data, in Div()
979 inline void Comparison(const T* input1_data, const Dims<4>& input1_dims, in Comparison() argument
984 ComparisonImpl<T, F>(op_params, DimsToShape(input1_dims), input1_data, in Comparison()
991 const Dims<4>& input1_dims, int32 input1_offset, in Comparison() argument
1008 ComparisonWithScaling<T, F>(op_params, DimsToShape(input1_dims), input1_data, in Comparison()
1015 const Dims<4>& input1_dims, in BroadcastComparison() argument
1021 BroadcastComparison4DSlowImpl<T, F>(op_params, DimsToShape(input1_dims), in BroadcastComparison()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dpack_test.cc55 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsFloat() local
63 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsFloat()
87 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackThreeInputsFloat() local
96 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackThreeInputsFloat()
122 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsQuantized() local
134 CreateQuantizedTensor(input1_data, input1_dims, 1.0, 128), in TestPackTwoInputsQuantized()
159 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsQuantized32() local
167 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsQuantized32()
Dmaximum_minimum_test.cc32 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinFloat() local
41 CreateTensor(input1_data, input1_dims), in TestMaxMinFloat()
73 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinQuantized() local
82 CreateQuantizedTensor(input1_data, input1_dims, input1_scale, in TestMaxMinQuantized()
112 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinQuantizedInt32() local
121 CreateTensor(input1_data, input1_dims), in TestMaxMinQuantizedInt32()
Dconcatenation_test.cc34 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestConcatenateTwoInputs() local
41 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestConcatenateTwoInputs()
77 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestConcatenateQuantizedTwoInputs() local
85 CreateQuantizedTensor(input1_data, input1_dims, input_scale, in TestConcatenateQuantizedTwoInputs()
Dmul_test.cc74 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMulFloat() local
83 CreateTensor(input1_data, input1_dims), in TestMulFloat()
101 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMulQuantized() local
110 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestMulQuantized()
Dcomparisons_test.cc58 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonFloat() local
63 CreateTensor(input1_data, input1_dims), in TestComparisonFloat()
76 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonBool() local
81 CreateTensor(input1_data, input1_dims), in TestComparisonBool()
94 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonInt() local
99 CreateTensor(input1_data, input1_dims), in TestComparisonInt()
115 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonQuantizedUInt8() local
120 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestComparisonQuantizedUInt8()
138 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonQuantizedInt8() local
143 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestComparisonQuantizedInt8()
Dadd_n.cc54 TfLiteIntArray* input1_dims = input1->dims; in Prepare() local
55 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input1_dims); in Prepare()
Dlogical_test.cc32 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestLogicalOp() local
41 CreateTensor(input1_data, input1_dims), in TestLogicalOp()
Dadd_test.cc94 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestAddFloat() local
102 CreateTensor(input1_data, input1_dims), in TestAddFloat()
121 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestAddQuantized() local
130 input1_dims, input1_scale, in TestAddQuantized()
Dsub_test.cc93 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestSubFloat() local
101 CreateTensor(input1_data, input1_dims), in TestSubFloat()
120 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestSubQuantized() local
129 input1_dims, input1_scale, in TestSubQuantized()
/external/tensorflow/tensorflow/lite/kernels/
Dadd_n.cc54 TfLiteIntArray* input1_dims = input1->dims; in Prepare() local
55 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input1_dims); in Prepare()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3436 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv() argument
3443 BroadcastDivSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastDiv()
3493 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add() argument
3502 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
3509 const Dims<4>& input1_dims, int32 input1_offset, in Add() argument
3541 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
3547 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add() argument
3556 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
3562 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd() argument
3569 BroadcastAdd4DSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastAdd()
[all …]
/external/tensorflow/tensorflow/lite/micro/
Dmemory_helpers_test.cc197 const int input1_dims[] = {1, 1}; in TF_LITE_MICRO_TEST() local
201 nullptr, tflite::testing::IntArrayFromInts(input1_dims)); in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h772 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast() argument
782 desc1_out->extents[i] = input1_dims.sizes[i]; in NdArrayDescsForElementwiseBroadcast()
783 desc1_out->strides[i] = input1_dims.strides[i]; in NdArrayDescsForElementwiseBroadcast()
791 const int extent1 = ArraySize(input1_dims, i); in NdArrayDescsForElementwiseBroadcast()