Home
last modified time | relevance | path

Searched defs:input1_shape (Results 1 – 25 of 32) sorted by relevance

12

/aosp_15_r20/external/XNNPACK/src/operators/
H A Dbinary-elementwise-nd.c843 const size_t* input1_shape, in setup_binary_elementwise_nd()
1053 const size_t* input1_shape, in setup_binary_elementwise_nd_f16()
1083 const size_t* input1_shape, in setup_binary_elementwise_nd_f32()
1107 const size_t* input1_shape, in xnn_setup_add_nd_f16()
1127 const size_t* input1_shape, in xnn_setup_add_nd_f32()
1147 const size_t* input1_shape, in xnn_setup_add_nd_qs8()
1170 const size_t* input1_shape, in xnn_setup_add_nd_qu8()
1193 const size_t* input1_shape, in xnn_setup_divide_nd_f16()
1213 const size_t* input1_shape, in xnn_setup_divide_nd_f32()
1233 const size_t* input1_shape, in xnn_setup_maximum_nd_f16()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dsub.h33 const RuntimeShape& input1_shape, in SubNonBroadcast()
49 const RuntimeShape& input1_shape, in SubNonBroadcast()
70 const RuntimeShape& input1_shape, in BroadcastSubSlow()
110 const RuntimeShape& input1_shape, in BroadcastSubSlow()
150 const RuntimeShape& input1_shape, in BroadcastSubSlow()
189 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastSubSlow()
226 const RuntimeShape& input1_shape, in BroadcastSub16POTSlow()
270 const RuntimeShape& input1_shape, in BroadcastQuantSubSlow()
354 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Sub()
370 const RuntimeShape& input1_shape, const int8_t* input1_data, in Sub()
[all …]
H A Dadd.h30 const RuntimeShape& input1_shape, const T* input1_data, in Add()
117 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Add()
133 const RuntimeShape& input1_shape, in AddGeneralParamScale()
200 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastAdd4DSlow()
245 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastAdd4DSlow()
H A Dmul.h49 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
66 const RuntimeShape& input1_shape, in Mul()
80 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Mul()
92 const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
H A Dreference_ops.h239 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
258 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
287 const RuntimeShape& input1_shape, const int16_t* input1_data, in Sub16()
708 void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum()
722 inline void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum()
730 void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum()
744 inline void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum()
752 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax()
762 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax()
843 inline void Pow(const RuntimeShape& input1_shape, const T* input1_data, in Pow()
H A Dcomparisons.h58 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonImpl()
70 const RuntimeShape& input1_shape, in Comparison()
81 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonWithScaling()
156 const RuntimeShape& input1_shape, in BroadcastComparison4DSlow()
H A Ddiv.h79 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Div()
91 const RuntimeShape& input1_shape, const int8_t* input1_data, in Div()
228 const RuntimeShape& input1_shape, const T* input1_data, in Div()
H A Darg_min_max.h36 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax()
78 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax()
H A Dbroadcast_args.h25 void BroadcastArgs(const RuntimeShape& input1_shape, const T* input1_data, in BroadcastArgs()
H A Dbatch_to_space_nd.h53 const RuntimeShape input1_shape = in BatchToSpaceND() local
H A Dspace_to_batch_nd.h56 const RuntimeShape input1_shape = in SpaceToBatchND() local
H A Dstring_comparisons.h41 const RuntimeShape& input1_shape, in ComparisonStringImpl()
H A Dbinary_function.h75 inline void BinaryFunction(const RuntimeShape& input1_shape, in BinaryFunction()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dmul.h48 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
62 const RuntimeShape& input1_shape, const int16_t* input1_data, in Mul()
92 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
H A Dadd.h50 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastBinaryFunction4DSlow()
119 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add()
131 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dsub.h125 const RuntimeShape& input1_shape, in BroadcastSubFiveFold()
194 const RuntimeShape& input1_shape, const int16* input1_data, in Sub()
211 const RuntimeShape& input1_shape, in BroadcastSubDispatch()
H A Dadd.h459 const RuntimeShape& input1_shape, const int8* input1_data, in Add()
476 const RuntimeShape& input1_shape, const int16* input1_data, in Add()
493 const RuntimeShape& input1_shape, in BroadcastAddDispatch()
H A Dmul.h233 const RuntimeShape& input1_shape, const int8* input1_data, in Mul()
246 const RuntimeShape& input1_shape, in BroadcastMulDispatch()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dset_ops.cc58 ShapeHandle input1_shape = c->input(1); in __anona97893f50102() local
76 ShapeHandle input1_shape = c->input(1); in __anona97893f50102() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dselect_test.cc33 BaseSelectOpModel(std::initializer_list<int> input1_shape, in BaseSelectOpModel()
62 SelectOpModel(std::initializer_list<int> input1_shape, in SelectOpModel()
75 SelectV2OpModel(std::initializer_list<int> input1_shape, in SelectV2OpModel()
H A Dlogical_test.cc31 LogicalOpModel(std::initializer_list<int> input1_shape, in LogicalOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/gradient/
H A Dbcast_grad_args.cc46 const RuntimeShape input1_shape = GetTensorShape(input1); in Prepare() local
72 const RuntimeShape input1_shape = GetTensorShape(input1); in Invoke() local
/aosp_15_r20/external/XNNPACK/test/
H A Dbinary-elementwise-operator-tester.h39 inline BinaryElementwiseOperatorTester& input1_shape(std::initializer_list<size_t> input1_shape) { in input1_shape() argument
45 inline const std::vector<size_t>& input1_shape() const { in input1_shape() function
H A Dsubgraph-binary-tester.h43 std::vector<size_t> input1_shape = RandomShape(); in SetUp() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h1518 const RuntimeShape& input1_shape, const float* input1_data, in Add()
1755 const RuntimeShape& input1_shape, const uint8* input1_data, in Add()
1772 const RuntimeShape& input1_shape, const int16* input1_data, in Add()
1810 const ArithmeticParams& params, const RuntimeShape& input1_shape, in Add()
1844 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastAddDispatch()
1943 const RuntimeShape& input1_shape, const float* input1_data, in Mul()
1954 const RuntimeShape& input1_shape, const int32* input1_data, in Mul()
1971 const RuntimeShape& input1_shape, in MulNoActivation()
1998 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
2019 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
[all …]

12