Home
last modified time | relevance | path

Searched defs:input2_shape (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsub.h35 const RuntimeShape& input2_shape, in SubNonBroadcast()
51 const RuntimeShape& input2_shape, in SubNonBroadcast()
72 const RuntimeShape& input2_shape, in BroadcastSubSlow()
112 const RuntimeShape& input2_shape, in BroadcastSubSlow()
169 const RuntimeShape& input2_shape, in BroadcastSubSlow()
209 const RuntimeShape& input2_shape, in BroadcastSubSlow()
263 const RuntimeShape& input2_shape, in BroadcastSubSlow()
301 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastSubSlow()
339 const RuntimeShape& input2_shape, in BroadcastSub16POTSlow()
448 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Sub()
[all …]
Dadd.h28 const RuntimeShape& input2_shape, const T* input2_data, in Add()
41 const RuntimeShape& input2_shape, const float* input2_data, in Add()
126 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Add()
143 const RuntimeShape& input2_shape, in AddGeneralParamScale()
208 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
248 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
292 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
Dmul.h47 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
64 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Mul()
77 const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
Dcomparisons.h59 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonImpl()
72 const RuntimeShape& input2_shape, in Comparison()
82 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonWithScaling()
158 const RuntimeShape& input2_shape, in BroadcastComparison4DSlow()
Ddiv.h63 const RuntimeShape& input2_shape, const uint8* input2_data, in Div()
176 const RuntimeShape& input2_shape, const T* input2_data, in Div()
Dbinary_function.h66 const RuntimeShape& input2_shape, in BinaryFunction()
Dstring_comparisons.h43 const RuntimeShape& input2_shape, in ComparisonStringImpl()
Dreference_ops.h222 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
241 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
270 const RuntimeShape& input2_shape, const int16_t* input2_data, in Sub16()
1477 const RuntimeShape& input2_shape, const T3* input2_data, in ArgMax()
1649 const RuntimeShape& input2_shape, const T* input2_data, in Pow()
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c378 const size_t* input2_shape, in setup_binary_elementwise_nd()
575 const size_t* input2_shape, in setup_binary_elementwise_nd_f16()
606 const size_t* input2_shape, in setup_binary_elementwise_nd_f32()
631 const size_t* input2_shape, in xnn_setup_add_nd_qs8()
655 const size_t* input2_shape, in xnn_setup_add_nd_f16()
675 const size_t* input2_shape, in xnn_setup_add_nd_f32()
695 const size_t* input2_shape, in xnn_setup_divide_nd_f32()
715 const size_t* input2_shape, in xnn_setup_maximum_nd_f32()
735 const size_t* input2_shape, in xnn_setup_minimum_nd_f32()
755 const size_t* input2_shape, in xnn_setup_multiply_nd_f16()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons.cc49 RuntimeShape input2_shape = tflite::micro::GetTensorShape(input2); in EqualEval() local
154 RuntimeShape input2_shape = tflite::micro::GetTensorShape(input2); in NotEqualEval() local
258 RuntimeShape input2_shape = tflite::micro::GetTensorShape(input2); in GreaterEval() local
349 RuntimeShape input2_shape = tflite::micro::GetTensorShape(input2); in GreaterEqualEval() local
440 RuntimeShape input2_shape = tflite::micro::GetTensorShape(input2); in LessEval() local
531 RuntimeShape input2_shape = tflite::micro::GetTensorShape(input2); in LessEqualEval() local
Dlogical_test.cc84 const int input2_shape[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
107 const int input2_shape[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
Dadd_test.cc205 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
377 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
424 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
Dsub_test.cc204 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
376 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
423 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
Dconcatenation_test.cc192 const int input2_shape[] = {3, 2, 3, 2}; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h47 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
61 const RuntimeShape& input2_shape, const int16_t* input2_data, in Mul()
91 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
Dadd.h50 const int8_t* input1_data, const RuntimeShape& input2_shape, in BroadcastBinaryFunction4DSlow()
119 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add()
132 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/lite/kernels/
Dselect_test.cc33 std::initializer_list<int> input2_shape, in BaseSelectOpModel()
62 std::initializer_list<int> input2_shape, in SelectOpModel()
75 std::initializer_list<int> input2_shape, in SelectV2OpModel()
Dlogical_test.cc32 std::initializer_list<int> input2_shape, BuiltinOperator op) { in LogicalOpModel()
Dcomparisons_test.cc36 std::initializer_list<int> input2_shape, in ComparisonOpModel()
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h82 inline BinaryElementwiseOperatorTester& input2_shape(std::initializer_list<size_t> input2_shape) { in input2_shape() function
88 inline const std::vector<size_t>& input2_shape() const { in input2_shape() function
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h234 const RuntimeShape& input2_shape, const int8* input2_data, in Mul()
248 const RuntimeShape& input2_shape, in BroadcastMulDispatch()
Dadd.h269 const RuntimeShape& input2_shape, const int8* input2_data, in Add()
287 const RuntimeShape& input2_shape, in BroadcastAddDispatch()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1850 const RuntimeShape& input2_shape, const float* input2_data, in Add()
2087 const RuntimeShape& input2_shape, const uint8* input2_data, in Add()
2104 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
2141 const RuntimeShape& input2_shape, const int32* input2_data, in Add()
2172 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastAddDispatch()
2271 const RuntimeShape& input2_shape, const float* input2_data, in Mul()
2282 const RuntimeShape& input2_shape, const int32* input2_data, in Mul()
2300 const RuntimeShape& input2_shape, in MulNoActivation()
2326 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
2347 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h853 const RuntimeShape& input2_shape, NdArrayDesc<N>* desc0_out, in NdArrayDescsForElementwiseBroadcast()