Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/XNNPACK/src/operators/
H A Dbinary-elementwise-nd.c845 const size_t* input2_shape, in setup_binary_elementwise_nd()
1055 const size_t* input2_shape, in setup_binary_elementwise_nd_f16()
1085 const size_t* input2_shape, in setup_binary_elementwise_nd_f32()
1109 const size_t* input2_shape, in xnn_setup_add_nd_f16()
1129 const size_t* input2_shape, in xnn_setup_add_nd_f32()
1149 const size_t* input2_shape, in xnn_setup_add_nd_qs8()
1172 const size_t* input2_shape, in xnn_setup_add_nd_qu8()
1195 const size_t* input2_shape, in xnn_setup_divide_nd_f16()
1215 const size_t* input2_shape, in xnn_setup_divide_nd_f32()
1235 const size_t* input2_shape, in xnn_setup_maximum_nd_f16()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A 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()
152 const RuntimeShape& input2_shape, in BroadcastSubSlow()
190 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastSubSlow()
228 const RuntimeShape& input2_shape, in BroadcastSub16POTSlow()
272 const RuntimeShape& input2_shape, in BroadcastQuantSubSlow()
355 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Sub()
371 const RuntimeShape& input2_shape, const int8_t* input2_data, in Sub()
[all …]
H A Dadd.h31 const RuntimeShape& input2_shape, const T* input2_data, in Add()
118 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Add()
135 const RuntimeShape& input2_shape, in AddGeneralParamScale()
201 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastAdd4DSlow()
246 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastAdd4DSlow()
H A Dmul.h50 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
68 const RuntimeShape& input2_shape, in Mul()
81 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Mul()
94 const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
H A 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()
H A Ddiv.h80 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Div()
92 const RuntimeShape& input2_shape, const int8_t* input2_data, in Div()
229 const RuntimeShape& input2_shape, const T* input2_data, in Div()
H A Dreference_ops.h240 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
259 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
288 const RuntimeShape& input2_shape, const int16_t* input2_data, in Sub16()
763 const RuntimeShape& input2_shape, const T3* input2_data, in ArgMax()
844 const RuntimeShape& input2_shape, const T* input2_data, in Pow()
H A Dbroadcast_args.h26 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastArgs()
H A Dstring_comparisons.h43 const RuntimeShape& input2_shape, in ComparisonStringImpl()
H A Dbinary_function.h77 const RuntimeShape& input2_shape, in BinaryFunction()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dmul.h49 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
63 const RuntimeShape& input2_shape, const int16_t* input2_data, in Mul()
93 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
H A Dadd.h51 const int8_t* input1_data, const RuntimeShape& input2_shape, in BroadcastBinaryFunction4DSlow()
120 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add()
133 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dsub.h127 const RuntimeShape& input2_shape, in BroadcastSubFiveFold()
195 const RuntimeShape& input2_shape, const int16* input2_data, in Sub()
213 const RuntimeShape& input2_shape, in BroadcastSubDispatch()
H A Dadd.h460 const RuntimeShape& input2_shape, const int8* input2_data, in Add()
477 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
495 const RuntimeShape& input2_shape, in BroadcastAddDispatch()
H A Dmul.h234 const RuntimeShape& input2_shape, const int8* input2_data, in Mul()
248 const RuntimeShape& input2_shape, in BroadcastMulDispatch()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dselect_test.cc34 std::initializer_list<int> input2_shape, in BaseSelectOpModel()
63 std::initializer_list<int> input2_shape, in SelectOpModel()
76 std::initializer_list<int> input2_shape, in SelectV2OpModel()
H A Dlogical_test.cc32 std::initializer_list<int> input2_shape, BuiltinOperator op) { in LogicalOpModel()
H A Dadd_test.cc45 const std::vector<int>& input2_shape, in BaseAddOpModel()
H A Dcomparisons_test.cc36 std::initializer_list<int> input2_shape, in ComparisonOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/gradient/
H A Dbcast_grad_args.cc53 const RuntimeShape input2_shape = GetTensorShape(input2); in Prepare() local
76 const RuntimeShape input2_shape = GetTensorShape(input2); in Invoke() local
/aosp_15_r20/external/XNNPACK/test/
H A Dbinary-elementwise-operator-tester.h81 inline BinaryElementwiseOperatorTester& input2_shape(std::initializer_list<size_t> input2_shape) { in input2_shape() function
87 inline const std::vector<size_t>& input2_shape() const { in input2_shape() function
H A Dsubgraph-binary-tester.h44 std::vector<size_t> input2_shape; in SetUp() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h1519 const RuntimeShape& input2_shape, const float* input2_data, in Add()
1756 const RuntimeShape& input2_shape, const uint8* input2_data, in Add()
1773 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
1811 const T* input1_data, const RuntimeShape& input2_shape, in Add()
1845 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastAddDispatch()
1944 const RuntimeShape& input2_shape, const float* input2_data, in Mul()
1955 const RuntimeShape& input2_shape, const int32* input2_data, in Mul()
1973 const RuntimeShape& input2_shape, in MulNoActivation()
1999 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
2020 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dcommon.h996 const RuntimeShape& input2_shape, NdArrayDesc<N>* desc0_out, in NdArrayDescsForElementwiseBroadcast()
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_quantized_api_test.cpp2088 const at::IntArrayRef input2_shape, in produce_inputs_for_binary_op()
2239 const at::IntArrayRef input2_shape, in test_quantized_binary_op()

12