Home
last modified time | relevance | path

Searched refs:input2_shape (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsub.h35 const RuntimeShape& input2_shape, in SubNonBroadcast() argument
40 MatchingElementsSize(input1_shape, input2_shape, output_shape); in SubNonBroadcast()
51 const RuntimeShape& input2_shape, in SubNonBroadcast() argument
56 MatchingElementsSize(input1_shape, input2_shape, output_shape); in SubNonBroadcast()
72 const RuntimeShape& input2_shape, in BroadcastSubSlow() argument
78 TFLITE_DCHECK_LE(input2_shape.DimensionsCount(), N); in BroadcastSubSlow()
83 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastSubSlow()
112 const RuntimeShape& input2_shape, in BroadcastSubSlow() argument
118 TFLITE_DCHECK_LE(input2_shape.DimensionsCount(), N); in BroadcastSubSlow()
123 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastSubSlow()
[all …]
Dcomparisons.h59 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonImpl() argument
62 MatchingFlatSize(input1_shape, input2_shape, output_shape); in ComparisonImpl()
72 const RuntimeShape& input2_shape, in Comparison() argument
75 ComparisonImpl<float, F>(op_params, input1_shape, input1_data, input2_shape, in Comparison()
82 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonWithScaling() argument
93 MatchingFlatSize(input1_shape, input2_shape, output_shape); in ComparisonWithScaling()
158 const RuntimeShape& input2_shape, in BroadcastComparison4DSlow() argument
163 input2_shape, input2_data, in BroadcastComparison4DSlow()
215 const RuntimeShape& input2_shape, const float* input2_data, \
217 Comparison<name##Fn>(op_params, input1_shape, input1_data, input2_shape, \
[all …]
Dadd.h31 const RuntimeShape& input2_shape, const T* input2_data, in Add() argument
37 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Add()
118 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Add() argument
123 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Add()
135 const RuntimeShape& input2_shape, in AddGeneralParamScale() argument
142 MatchingElementsSize(input1_shape, input2_shape, output_shape); in AddGeneralParamScale()
155 const RuntimeShape& input2_shape, const int16_t* input2_data,
159 AddGeneralParamScale(params, input1_shape, input1_data, input2_shape,
169 MatchingElementsSize(input1_shape, input2_shape, output_shape);
201 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastAdd4DSlow() argument
[all …]
Dmul.h50 const RuntimeShape& input2_shape, const T* input2_data, in Mul() argument
57 MatchingExtendedShapeFlatSize(input1_shape, input2_shape, output_shape); in Mul()
68 const RuntimeShape& input2_shape, in Mul() argument
73 MatchingExtendedShapeFlatSize(input1_shape, input2_shape, output_shape); in Mul()
81 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Mul() argument
86 MatchingExtendedShapeFlatSize(input1_shape, input2_shape, output_shape); in Mul()
94 const RuntimeShape& input2_shape, in BroadcastMul4DSlow() argument
100 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastMul4DSlow()
Ddiv.h80 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Div() argument
85 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Div()
92 const RuntimeShape& input2_shape, const int8_t* input2_data, in Div() argument
97 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Div()
229 const RuntimeShape& input2_shape, const T* input2_data, in Div() argument
236 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Div()
Dbroadcast_args.h26 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastArgs() argument
41 int shape2_i = get_shape_data(input2_shape, input2_data, i); in BroadcastArgs()
Dstring_comparisons.h43 const RuntimeShape& input2_shape, in ComparisonStringImpl() argument
48 MatchingFlatSize(input1_shape, input2_shape, output_shape); in ComparisonStringImpl()
Dbinary_function.h77 const RuntimeShape& input2_shape, in BinaryFunction() argument
82 MatchingFlatSize(input1_shape, input2_shape, output_shape); in BinaryFunction()
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c845 const size_t* input2_shape, in setup_binary_elementwise_nd() argument
895 const size_t input2_dim = input2_shape[num_input2_dims - i]; in setup_binary_elementwise_nd()
953 const size_t input2_dim = input2_shape[i]; in setup_binary_elementwise_nd()
1055 const size_t* input2_shape, in setup_binary_elementwise_nd_f16() argument
1068 input2_shape, in setup_binary_elementwise_nd_f16()
1085 const size_t* input2_shape, in setup_binary_elementwise_nd_f32() argument
1095 num_input2_dims, input2_shape, in setup_binary_elementwise_nd_f32()
1109 const size_t* input2_shape, in xnn_setup_add_nd_f16() argument
1118 num_input2_dims, input2_shape, in xnn_setup_add_nd_f16()
1129 const size_t* input2_shape, in xnn_setup_add_nd_f32() argument
[all …]
/external/XNNPACK/test/
Dsubgraph-binary-tester.h44 std::vector<size_t> input2_shape; in SetUp() local
49 input2_shape = RandomShape(input2_num_dims); in SetUp()
51 std::copy_backward(input1_shape.begin(), input1_shape.end(), input2_shape.end()); in SetUp()
58 …const size_t input2_broadcast_dim = std::uniform_int_distribution<size_t>(0, input2_shape.size())(… in SetUp()
59 if (input2_broadcast_dim < input2_shape.size()) { in SetUp()
60 input2_shape[input2_broadcast_dim] = 1; in SetUp()
67 std::copy_backward(input2_shape.cbegin(), input2_shape.cend(), input2_dims.end()); in SetUp()
76 input2 = std::vector<T>(XNN_EXTRA_BYTES / sizeof(T) + NumElements(input2_shape)); in SetUp()
Dadd-nd.cc41 .input2_shape({input2_dim1})
56 .input2_shape({input2_dim1})
70 .input2_shape({input2_dim2, input2_dim1})
87 .input2_shape({input2_dim2, input2_dim1})
118 .input2_shape({input2_dim1})
138 .input2_shape({input2_dim2, input2_dim1})
154 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
173 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
195 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
230 .input2_shape({input2_dim1})
[all …]
Dmultiply-nd.cc41 .input2_shape({input2_dim1})
56 .input2_shape({input2_dim1})
70 .input2_shape({input2_dim2, input2_dim1})
87 .input2_shape({input2_dim2, input2_dim1})
118 .input2_shape({input2_dim1})
138 .input2_shape({input2_dim2, input2_dim1})
154 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
173 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
195 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
230 .input2_shape({input2_dim1})
[all …]
Dsubtract-nd.cc41 .input2_shape({input2_dim1})
56 .input2_shape({input2_dim1})
70 .input2_shape({input2_dim2, input2_dim1})
87 .input2_shape({input2_dim2, input2_dim1})
118 .input2_shape({input2_dim1})
138 .input2_shape({input2_dim2, input2_dim1})
154 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
173 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
195 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
230 .input2_shape({input2_dim1})
[all …]
Ddivide-nd.cc42 .input2_shape({input2_dim1})
57 .input2_shape({input2_dim1})
71 .input2_shape({input2_dim2, input2_dim1})
88 .input2_shape({input2_dim2, input2_dim1})
119 .input2_shape({input2_dim1})
139 .input2_shape({input2_dim2, input2_dim1})
155 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
174 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
196 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
231 .input2_shape({input2_dim1})
[all …]
Dminimum-nd.cc42 .input2_shape({input2_dim1})
57 .input2_shape({input2_dim1})
71 .input2_shape({input2_dim2, input2_dim1})
88 .input2_shape({input2_dim2, input2_dim1})
119 .input2_shape({input2_dim1})
139 .input2_shape({input2_dim2, input2_dim1})
155 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
174 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
196 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
231 .input2_shape({input2_dim1})
[all …]
Dmaximum-nd.cc42 .input2_shape({input2_dim1})
57 .input2_shape({input2_dim1})
71 .input2_shape({input2_dim2, input2_dim1})
88 .input2_shape({input2_dim2, input2_dim1})
119 .input2_shape({input2_dim1})
139 .input2_shape({input2_dim2, input2_dim1})
155 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
174 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
196 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
231 .input2_shape({input2_dim1})
[all …]
Dsquared-difference-nd.cc42 .input2_shape({input2_dim1})
57 .input2_shape({input2_dim1})
71 .input2_shape({input2_dim2, input2_dim1})
88 .input2_shape({input2_dim2, input2_dim1})
119 .input2_shape({input2_dim1})
139 .input2_shape({input2_dim2, input2_dim1})
155 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
174 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
196 .input2_shape({input2_dim3, input2_dim2, input2_dim1})
231 .input2_shape({input2_dim1})
[all …]
Dbinary-elementwise-operator-tester.h81 inline BinaryElementwiseOperatorTester& input2_shape(std::initializer_list<size_t> input2_shape) { in input2_shape() function
82 assert(input2_shape.size() <= XNN_MAX_TENSOR_DIMS); in input2_shape()
83 this->input2_shape_ = std::vector<size_t>(input2_shape); in input2_shape()
87 inline const std::vector<size_t>& input2_shape() const { in input2_shape() function
221 … std::copy(input2_shape().cbegin(), input2_shape().cend(), input2_dims.end() - num_input2_dims()); in TestQS8()
325 input2_shape().data(), in TestQS8()
336 input2_shape().data(), in TestQS8()
347 input2_shape().data(), in TestQS8()
403 … std::copy(input2_shape().cbegin(), input2_shape().cend(), input2_dims.end() - num_input2_dims()); in TestQU8()
507 input2_shape().data(), in TestQU8()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h49 const RuntimeShape& input2_shape, const T* input2_data, in Mul() argument
55 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Mul()
63 const RuntimeShape& input2_shape, const int16_t* input2_data, in Mul() argument
72 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Mul()
93 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastMul4DSlow() argument
100 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastMul4DSlow()
Dadd.h51 const int8_t* input1_data, const RuntimeShape& input2_shape, in BroadcastBinaryFunction4DSlow() argument
58 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastBinaryFunction4DSlow()
120 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add() argument
125 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Add()
133 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow() argument
137 BroadcastBinaryFunction4DSlow(params, input1_shape, input1_data, input2_shape, in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dsub.h127 const RuntimeShape& input2_shape, in BroadcastSubFiveFold() argument
195 const RuntimeShape& input2_shape, const int16* input2_data, in Sub() argument
206 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Sub()
213 const RuntimeShape& input2_shape, in BroadcastSubDispatch() argument
227 params, input1_shape, input1_data, input2_shape, input2_data, in BroadcastSubDispatch()
231 BroadcastSubFiveFold(params, input1_shape, input1_data, input2_shape, in BroadcastSubDispatch()
Dmul.h234 const RuntimeShape& input2_shape, const int8* input2_data, in Mul() argument
240 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Mul()
248 const RuntimeShape& input2_shape, in BroadcastMulDispatch() argument
254 params, input1_shape, input1_data, input2_shape, input2_data, in BroadcastMulDispatch()
259 params, input1_shape, input1_data, input2_shape, input2_data, in BroadcastMulDispatch()
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args.cc53 const RuntimeShape input2_shape = GetTensorShape(input2); in Prepare() local
55 TF_LITE_ENSURE_EQ(context, input2_shape.DimensionsCount(), 1); in Prepare()
76 const RuntimeShape input2_shape = GetTensorShape(input2); in Invoke() local
94 input2->data.i32 + input2_shape.Dims(0)); in Invoke()
97 input2->data.i64 + input2_shape.Dims(0)); in Invoke()
/external/tensorflow/tensorflow/lite/kernels/
Dselect_test.cc34 std::initializer_list<int> input2_shape, in BaseSelectOpModel() argument
63 std::initializer_list<int> input2_shape, in SelectOpModel() argument
65 : BaseSelectOpModel(input1_shape, input2_shape, input3_shape, in SelectOpModel()
69 BuildInterpreter({input1_shape, input2_shape, input3_shape}); in SelectOpModel()
76 std::initializer_list<int> input2_shape, in SelectV2OpModel() argument
79 : BaseSelectOpModel(input1_shape, input2_shape, input3_shape, in SelectV2OpModel()
83 BuildInterpreter({input1_shape, input2_shape, input3_shape}); in SelectV2OpModel()
Dlogical_test.cc32 std::initializer_list<int> input2_shape, BuiltinOperator op) { in LogicalOpModel() argument
37 BuildInterpreter({input1_shape, input2_shape}); in LogicalOpModel()

12