| /external/XNNPACK/src/ |
| D | binary-elementwise-nd.c | 143 const size_t* input2_shape, in setup_binary_elementwise_nd_f32()
311 const size_t* input2_shape, in xnn_setup_add_nd_f32()
331 const size_t* input2_shape, in xnn_setup_divide_nd_f32()
351 const size_t* input2_shape, in xnn_setup_maximum_nd_f32()
371 const size_t* input2_shape, in xnn_setup_minimum_nd_f32()
391 const size_t* input2_shape, in xnn_setup_multiply_nd_f32()
411 const size_t* input2_shape, in xnn_setup_subtract_nd_f32()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | mul.h | 45 const RuntimeShape& input2_shape, const int8_t* input2_data, in Mul()
59 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
89 const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
|
| D | add.h | 62 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add()
80 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | add.h | 28 const RuntimeShape& input2_shape, const T* input2_data, in Add()
41 const RuntimeShape& input2_shape, const float* input2_data, in Add()
121 const RuntimeShape& input2_shape, const uint8* input2_data, in Add()
137 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
182 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
222 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
263 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
|
| D | mul.h | 47 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
64 const RuntimeShape& input2_shape, const uint8* input2_data, in Mul()
77 const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
|
| D | comparisons.h | 58 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonImpl()
71 const RuntimeShape& input2_shape, in Comparison()
81 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonWithScaling()
141 const RuntimeShape& input2_shape, in BroadcastComparison4DSlow()
|
| D | binary_function.h | 70 const RuntimeShape& input2_shape, in BinaryFunction()
|
| D | reference_ops.h | 428 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
447 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
530 const RuntimeShape& input2_shape, const T* input2_data, in Div()
582 const RuntimeShape& input2_shape, const uint8* input2_data, in Div()
656 const RuntimeShape& input2_shape, in SubNonBroadcast()
672 const RuntimeShape& input2_shape, in SubNonBroadcast()
696 const RuntimeShape& input2_shape, in BroadcastSub4DSlow()
737 const RuntimeShape& input2_shape, in BroadcastSub4DSlow()
801 const RuntimeShape& input2_shape, in BroadcastSub4DSlow()
843 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastSub4DSlow()
[all …]
|
| /external/XNNPACK/test/ |
| D | binary-elementwise-operator-tester.h | 59 inline BinaryElementwiseOperatorTester& input2_shape(std::initializer_list<size_t> input2_shape) { in input2_shape() function
65 inline const std::vector<size_t>& input2_shape() const { in input2_shape() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | select_test.cc | 29 std::initializer_list<int> input2_shape, in BaseSelectOpModel()
58 std::initializer_list<int> input2_shape, in SelectOpModel()
71 std::initializer_list<int> input2_shape, in SelectV2OpModel()
|
| D | logical_test.cc | 29 std::initializer_list<int> input2_shape, BuiltinOperator op) { in LogicalOpModel()
|
| D | comparisons_test.cc | 29 std::initializer_list<int> input2_shape, in ComparisonOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | mul.h | 173 const RuntimeShape& input2_shape, const int8* input2_data, in Mul()
258 const RuntimeShape& input2_shape, in BroadcastMulDispatch()
|
| D | add.h | 219 const RuntimeShape& input2_shape, const int8* input2_data, in Add()
332 const RuntimeShape& input2_shape, in BroadcastAddDispatch()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | add_test.cc | 237 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
409 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
462 const int input2_shape[] = {0}; in TF_LITE_MICRO_TEST() local
|
| D | concatenation_test.cc | 234 auto input2_shape = {3, 2, 3, 2}; in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1717 const RuntimeShape& input2_shape, const float* input2_data, in Add()
1954 const RuntimeShape& input2_shape, const uint8* input2_data, in Add()
1971 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
2008 const RuntimeShape& input2_shape, const int32* input2_data, in Add()
2216 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastAddDispatch()
2286 const RuntimeShape& input2_shape, const float* input2_data, in Mul()
2297 const RuntimeShape& input2_shape, const int32* input2_data, in Mul()
2315 const RuntimeShape& input2_shape, in MulNoActivation()
2341 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
2362 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | common.h | 632 const RuntimeShape& input2_shape, NdArrayDesc<N>* desc0_out, in NdArrayDescsForElementwiseBroadcast()
|