Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h609 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv() argument
616 BroadcastDiv4DSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastDiv()
622 inline void Div(const T* input1_data, const Dims<4>& input1_dims, in Div() argument
629 Div(op_params, DimsToShape(input1_dims), input1_data, in Div()
1009 inline void Comparison(const T* input1_data, const Dims<4>& input1_dims, in Comparison() argument
1014 ComparisonImpl<T, F>(op_params, DimsToShape(input1_dims), input1_data, in Comparison()
1021 const Dims<4>& input1_dims, int32 input1_offset, in Comparison() argument
1038 ComparisonWithScaling<T, F>(op_params, DimsToShape(input1_dims), input1_data, in Comparison()
1045 const Dims<4>& input1_dims, in BroadcastComparison() argument
1051 BroadcastComparison4DSlowImpl<T, F>(op_params, DimsToShape(input1_dims), in BroadcastComparison()
[all …]
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h145 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestF32() local
148 std::fill(input1_dims.begin(), input1_dims.end(), 1); in TestF32()
151 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestF32()
154 if (input1_dims[i] != 1 && input2_dims[i] != 1) { in TestF32()
155 ASSERT_EQ(input1_dims[i], input2_dims[i]); in TestF32()
157 output_dims[i] = std::max(input1_dims[i], input2_dims[i]); in TestF32()
168 input1_strides[i - 1] = input1_dims[i - 1] == 1 ? 0 : input1_stride; in TestF32()
171 input1_stride *= input1_dims[i - 1]; in TestF32()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dpack_test.cc33 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestPackTwoInputsFloat() local
42 CreateFloatTensor(input1_data, input1_dims, "input1_tensor"), in TestPackTwoInputsFloat()
107 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestPackThreeInputsFloat() local
117 CreateFloatTensor(input1_data, input1_dims, "input1_tensor"), in TestPackThreeInputsFloat()
181 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestPackTwoInputsQuantized() local
192 CreateQuantizedTensor(input1_data, input1_dims, "input1_tensor", 0, 10), in TestPackTwoInputsQuantized()
253 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestPackTwoInputsQuantized32() local
262 CreateQuantized32Tensor(input1_data, input1_dims, "input1_tensor", 1.0), in TestPackTwoInputsQuantized32()
Dmaximum_minimum_test.cc34 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestMaxMinFloat() local
43 CreateFloatTensor(input1_data, input1_dims, "input1_tensor"), in TestMaxMinFloat()
90 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestMaxMinQuantized() local
99 CreateQuantizedTensor(input1_data, input1_dims, "input1_tensor", in TestMaxMinQuantized()
147 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestMaxMinQuantizedInt32() local
156 CreateQuantized32Tensor(input1_data, input1_dims, "input1_tensor", in TestMaxMinQuantizedInt32()
Dconcatenation_test.cc35 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestConcatenateTwoInputs() local
43 CreateFloatTensor(input1_data, input1_dims, "input1_tensor"), in TestConcatenateTwoInputs()
94 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestConcatenateQuantizedTwoInputs() local
102 CreateQuantizedTensor(input1_data, input1_dims, "input1_tensor", in TestConcatenateQuantizedTwoInputs()
Dcomparisons_test.cc76 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonFloat() local
81 CreateFloatTensor(input1_data, input1_dims, "input1_tensor"), in TestComparisonFloat()
93 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonBool() local
98 CreateBoolTensor(input1_data, input1_dims, "input1_tensor"), in TestComparisonBool()
110 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonInt() local
115 CreateInt32Tensor(input1_data, input1_dims, "input1_tensor"), in TestComparisonInt()
131 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonQuantizedUInt8() local
137 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestComparisonQuantizedUInt8()
155 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonQuantizedInt8() local
161 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestComparisonQuantizedInt8()
Dlogical_test.cc33 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestLogicalOp() local
42 CreateBoolTensor(input1_data, input1_dims, "input1_tensor"), in TestLogicalOp()
Dmul_test.cc33 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestMulFloat() local
44 CreateFloatTensor(input1_data, input1_dims, "input1_tensor"), in TestMulFloat()
105 TfLiteIntArray* input1_dims = IntArrayFromInitializer(input1_dims_data); in TestMulQuantized() local
116 CreateQuantizedTensor(input1_data, input1_dims, "input1_tensor", in TestMulQuantized()
Dadd_test.cc124 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestAddFloat() local
132 CreateFloatTensor(input1_data, input1_dims, "input1_tensor"), in TestAddFloat()
151 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestAddQuantized() local
160 input1_data, input1_quantized, input1_dims, input1_scale, in TestAddQuantized()
/external/tensorflow/tensorflow/lite/kernels/
Dadd_n.cc47 TfLiteIntArray* input1_dims = input1->dims; in Prepare() local
48 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input1_dims); in Prepare()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3423 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv() argument
3430 BroadcastDiv4DSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastDiv()
3480 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add() argument
3489 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
3496 const Dims<4>& input1_dims, int32 input1_offset, in Add() argument
3528 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
3534 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add() argument
3543 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
3549 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd() argument
3556 BroadcastAdd4DSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastAdd()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h551 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast() argument
561 desc1_out->extents[i] = input1_dims.sizes[i]; in NdArrayDescsForElementwiseBroadcast()
562 desc1_out->strides[i] = input1_dims.strides[i]; in NdArrayDescsForElementwiseBroadcast()
570 const int extent1 = ArraySize(input1_dims, i); in NdArrayDescsForElementwiseBroadcast()