Home
last modified time | relevance | path

Searched refs:input1_shape (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons.cc48 RuntimeShape input1_shape = tflite::micro::GetTensorShape(input1); in EqualEval() local
58 data->params, input1_shape, in EqualEval()
63 data->params, input1_shape, in EqualEval()
71 data->params, input1_shape, in EqualEval()
76 data->params, input1_shape, in EqualEval()
84 data->params, input1_shape, in EqualEval()
89 data->params, input1_shape, in EqualEval()
97 data->params, input1_shape, in EqualEval()
102 data->params, input1_shape, in EqualEval()
110 data->params, input1_shape, in EqualEval()
[all …]
Dlogical_test.cc82 const int input1_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
89 input1_shape, input1, input2_shape, input2, in TF_LITE_MICRO_TEST()
90 input1_shape, golden, output_data); in TF_LITE_MICRO_TEST()
105 const int input1_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
112 input1_shape, input1, input2_shape, input2, in TF_LITE_MICRO_TEST()
113 input1_shape, golden, output_data); in TF_LITE_MICRO_TEST()
Darg_min_max.cc35 inline void ArgMinMaxHelper(const RuntimeShape& input1_shape, in ArgMinMaxHelper() argument
40 reference_ops::ArgMinMax(input1_shape, input1_data, input2_data, in ArgMinMaxHelper()
43 reference_ops::ArgMinMax(input1_shape, input1_data, input2_data, in ArgMinMaxHelper()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Darg_min_max.h25 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax() argument
28 TFLITE_DCHECK_GT(input1_shape.DimensionsCount(), 0); in ArgMinMax()
29 TFLITE_DCHECK_EQ(input1_shape.DimensionsCount() - 1, in ArgMinMax()
33 axis += input1_shape.DimensionsCount(); in ArgMinMax()
35 const int axis_size = input1_shape.Dims(axis); in ArgMinMax()
39 TFLITE_DCHECK_EQ(input1_shape.Dims(i), output_shape.Dims(i)); in ArgMinMax()
40 outer_size *= input1_shape.Dims(i); in ArgMinMax()
44 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax()
46 TFLITE_DCHECK_EQ(input1_shape.Dims(i), output_shape.Dims(i - 1)); in ArgMinMax()
47 inner_size *= input1_shape.Dims(i); in ArgMinMax()
Dcomparisons.h58 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonImpl() argument
62 MatchingFlatSize(input1_shape, input2_shape, output_shape); in ComparisonImpl()
70 const RuntimeShape& input1_shape, in Comparison() argument
75 ComparisonImpl<float, F>(op_params, input1_shape, input1_data, input2_shape, in Comparison()
81 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonWithScaling() argument
93 MatchingFlatSize(input1_shape, input2_shape, output_shape); in ComparisonWithScaling()
156 const RuntimeShape& input1_shape, in BroadcastComparison4DSlow() argument
162 BroadcastComparison4DSlowImpl<float, F>(op_params, input1_shape, input1_data, in BroadcastComparison4DSlow()
214 const RuntimeShape& input1_shape, const float* input1_data, \
217 Comparison<name##Fn>(op_params, input1_shape, input1_data, input2_shape, \
[all …]
Dsub.h33 const RuntimeShape& input1_shape, in SubNonBroadcast() argument
40 MatchingElementsSize(input1_shape, input2_shape, output_shape); in SubNonBroadcast()
49 const RuntimeShape& input1_shape, in SubNonBroadcast() argument
56 MatchingElementsSize(input1_shape, input2_shape, output_shape); in SubNonBroadcast()
70 const RuntimeShape& input1_shape, in BroadcastSubSlow() argument
77 TFLITE_DCHECK_LE(input1_shape.DimensionsCount(), N); in BroadcastSubSlow()
83 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastSubSlow()
110 const RuntimeShape& input1_shape, in BroadcastSubSlow() argument
117 TFLITE_DCHECK_LE(input1_shape.DimensionsCount(), N); in BroadcastSubSlow()
123 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastSubSlow()
[all …]
Dbatch_to_space_nd.h53 const RuntimeShape input1_shape = in BatchToSpaceND() local
62 const int depth = input1_shape.Dims(3); in BatchToSpaceND()
63 const int input_width = input1_shape.Dims(2); in BatchToSpaceND()
64 const int input_height = input1_shape.Dims(1); in BatchToSpaceND()
65 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND()
91 input1_data + Offset(input1_shape, in_batch, in_h, in_w, 0); in BatchToSpaceND()
Dspace_to_batch_nd.h56 const RuntimeShape input1_shape = in SpaceToBatchND() local
61 const int depth = input1_shape.Dims(3); in SpaceToBatchND()
62 const int input_width = input1_shape.Dims(2); in SpaceToBatchND()
63 const int input_height = input1_shape.Dims(1); in SpaceToBatchND()
64 const int input_batch_size = input1_shape.Dims(0); in SpaceToBatchND()
96 Offset(input1_shape, input_batch, in SpaceToBatchND()
Dadd.h27 const RuntimeShape& input1_shape, const T* input1_data, in Add() argument
31 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Add()
40 const RuntimeShape& input1_shape, const float* input1_data, in Add() argument
44 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Add()
125 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Add() argument
131 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Add()
141 const RuntimeShape& input1_shape, in AddGeneralParamScale() argument
150 MatchingElementsSize(input1_shape, input2_shape, output_shape); in AddGeneralParamScale()
162 const RuntimeShape& input1_shape, const int16_t* input1_data,
167 AddGeneralParamScale(params, input1_shape, input1_data, input2_shape,
[all …]
Dmul.h46 const RuntimeShape& input1_shape, const T* input1_data, in Mul() argument
54 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
63 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Mul() argument
69 MatchingFlatSize(input1_shape, input2_shape, output_shape); in Mul()
75 const RuntimeShape& input1_shape, in BroadcastMul4DSlow() argument
83 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastMul4DSlow()
Ddiv.h62 const RuntimeShape& input1_shape, const uint8* input1_data, in Div() argument
68 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Div()
175 const RuntimeShape& input1_shape, const T* input1_data, in Div() argument
183 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Div()
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c376 const size_t* input1_shape, in setup_binary_elementwise_nd() argument
421 if (input1_shape[i] == 0) { in setup_binary_elementwise_nd()
452 const size_t input1_dim = input1_shape[num_input1_dims - i]; in setup_binary_elementwise_nd()
499 const size_t input1_dim = input1_shape[i]; in setup_binary_elementwise_nd()
573 const size_t* input1_shape, in setup_binary_elementwise_nd_f16() argument
586 input1_shape, in setup_binary_elementwise_nd_f16()
604 const size_t* input1_shape, in setup_binary_elementwise_nd_f32() argument
615 num_input1_dims, input1_shape, in setup_binary_elementwise_nd_f32()
629 const size_t* input1_shape, in xnn_setup_add_nd_qs8() argument
639 num_input1_dims, input1_shape, in xnn_setup_add_nd_qs8()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dset_ops.cc58 ShapeHandle input1_shape = c->input(1); in __anon9b840f3f0102() local
60 c->WithRank(input1_shape, input0_rank, &input1_shape)); in __anon9b840f3f0102()
61 if (c->RankKnown(input1_shape)) { in __anon9b840f3f0102()
63 const int32 rank = c->Rank(input1_shape); in __anon9b840f3f0102()
69 c->Subshape(input1_shape, 0, rank - 1, &group1_shape)); in __anon9b840f3f0102()
76 ShapeHandle input1_shape = c->input(1); in __anon9b840f3f0102() local
77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input1_shape, 2, &input1_shape)); in __anon9b840f3f0102()
78 if (c->RankKnown(input1_shape)) { in __anon9b840f3f0102()
79 output_rank = c->MakeDim(c->Rank(input1_shape)); in __anon9b840f3f0102()
/external/XNNPACK/test/
Dmultiply-nd.cc30 .input1_shape({input1_dim1})
55 .input1_shape({input1_dim1})
86 .input1_shape({input1_dim1})
101 .input1_shape({input1_dim2, input1_dim1})
117 .input1_shape({input1_dim2, input1_dim1})
137 .input1_shape({input1_dim2, input1_dim1})
172 .input1_shape({input1_dim1})
194 .input1_shape({input1_dim2, input1_dim1})
211 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
229 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
[all …]
Ddivide-nd.cc30 .input1_shape({input1_dim1})
55 .input1_shape({input1_dim1})
86 .input1_shape({input1_dim1})
101 .input1_shape({input1_dim2, input1_dim1})
117 .input1_shape({input1_dim2, input1_dim1})
137 .input1_shape({input1_dim2, input1_dim1})
172 .input1_shape({input1_dim1})
194 .input1_shape({input1_dim2, input1_dim1})
211 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
229 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
[all …]
Dminimum-nd.cc30 .input1_shape({input1_dim1})
55 .input1_shape({input1_dim1})
86 .input1_shape({input1_dim1})
101 .input1_shape({input1_dim2, input1_dim1})
117 .input1_shape({input1_dim2, input1_dim1})
137 .input1_shape({input1_dim2, input1_dim1})
172 .input1_shape({input1_dim1})
194 .input1_shape({input1_dim2, input1_dim1})
211 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
229 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
[all …]
Dmaximum-nd.cc30 .input1_shape({input1_dim1})
55 .input1_shape({input1_dim1})
86 .input1_shape({input1_dim1})
101 .input1_shape({input1_dim2, input1_dim1})
117 .input1_shape({input1_dim2, input1_dim1})
137 .input1_shape({input1_dim2, input1_dim1})
172 .input1_shape({input1_dim1})
194 .input1_shape({input1_dim2, input1_dim1})
211 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
229 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
[all …]
Dsquared-difference-nd.cc30 .input1_shape({input1_dim1})
55 .input1_shape({input1_dim1})
86 .input1_shape({input1_dim1})
101 .input1_shape({input1_dim2, input1_dim1})
117 .input1_shape({input1_dim2, input1_dim1})
137 .input1_shape({input1_dim2, input1_dim1})
172 .input1_shape({input1_dim1})
194 .input1_shape({input1_dim2, input1_dim1})
211 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
229 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
[all …]
Dsubtract-nd.cc30 .input1_shape({input1_dim1})
55 .input1_shape({input1_dim1})
86 .input1_shape({input1_dim1})
101 .input1_shape({input1_dim2, input1_dim1})
117 .input1_shape({input1_dim2, input1_dim1})
137 .input1_shape({input1_dim2, input1_dim1})
172 .input1_shape({input1_dim1})
194 .input1_shape({input1_dim2, input1_dim1})
211 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
229 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
[all …]
Dadd-nd.cc30 .input1_shape({input1_dim1})
55 .input1_shape({input1_dim1})
86 .input1_shape({input1_dim1})
101 .input1_shape({input1_dim2, input1_dim1})
117 .input1_shape({input1_dim2, input1_dim1})
137 .input1_shape({input1_dim2, input1_dim1})
172 .input1_shape({input1_dim1})
194 .input1_shape({input1_dim2, input1_dim1})
211 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
229 .input1_shape({input1_dim3, input1_dim2, input1_dim1})
[all …]
Dbinary-elementwise-operator-tester.h40 inline BinaryElementwiseOperatorTester& input1_shape(std::initializer_list<size_t> input1_shape) { in input1_shape() function
41 assert(input1_shape.size() <= XNN_MAX_TENSOR_DIMS); in input1_shape()
42 this->input1_shape_ = std::vector<size_t>(input1_shape); in input1_shape()
46 inline const std::vector<size_t>& input1_shape() const { in input1_shape() function
221 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestQS8()
308 input1_shape().data(), in TestQS8()
359 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestF16()
452 input1_shape().data(), in TestF16()
463 input1_shape().data(), in TestF16()
510 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestF32()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h46 const RuntimeShape& input1_shape, const T* input1_data, in Mul() argument
53 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Mul()
60 const RuntimeShape& input1_shape, const int16_t* input1_data, in Mul() argument
70 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Mul()
90 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastMul4DSlow() argument
98 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastMul4DSlow()
Dadd.h49 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastBinaryFunction4DSlow() argument
57 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1, in BroadcastBinaryFunction4DSlow()
118 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add() argument
124 MatchingElementsSize(input1_shape, input2_shape, output_shape); in Add()
130 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow() argument
136 BroadcastBinaryFunction4DSlow(params, input1_shape, input1_data, input2_shape, in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc100 Shape input1_shape = input1_array.shape(); in EvaluateBinaryOperatorOnConstantInputs() local
102 ExtendShape(&input1_shape, dims_count); in EvaluateBinaryOperatorOnConstantInputs()
107 CHECK_EQ(input0_shape.dimensions_count(), input1_shape.dimensions_count()); in EvaluateBinaryOperatorOnConstantInputs()
110 VectorGreaterThan(input0_shape.dims(), input1_shape.dims()); in EvaluateBinaryOperatorOnConstantInputs()
114 PairwiseVectorSelect(input0_larger, input0_shape.dims(), input1_shape.dims(), in EvaluateBinaryOperatorOnConstantInputs()
134 const auto val1 = input1_data[Offset(input1_shape, input1_indices)]; in EvaluateBinaryOperatorOnConstantInputs()
/external/tensorflow/tensorflow/lite/kernels/
Dselect_test.cc32 BaseSelectOpModel(std::initializer_list<int> input1_shape, in BaseSelectOpModel() argument
61 SelectOpModel(std::initializer_list<int> input1_shape, in SelectOpModel() argument
64 : BaseSelectOpModel(input1_shape, input2_shape, input3_shape, in SelectOpModel()
68 BuildInterpreter({input1_shape, input2_shape, input3_shape}); in SelectOpModel()
74 SelectV2OpModel(std::initializer_list<int> input1_shape, in SelectV2OpModel() argument
78 : BaseSelectOpModel(input1_shape, input2_shape, input3_shape, in SelectV2OpModel()
82 BuildInterpreter({input1_shape, input2_shape, input3_shape}); in SelectV2OpModel()

12