Home
last modified time | relevance | path

Searched defs:input2_shape (Results 1 – 22 of 22) 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 …]
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()
Dadd.h30 const RuntimeShape& input2_shape, const T* input2_data, in Add()
117 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Add()
134 const RuntimeShape& input2_shape, in AddGeneralParamScale()
200 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastAdd4DSlow()
245 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastAdd4DSlow()
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.h76 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Div()
88 const RuntimeShape& input2_shape, const int8_t* input2_data, in Div()
221 const RuntimeShape& input2_shape, const T* input2_data, in Div()
Dbroadcast_args.h26 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastArgs()
Dbinary_function.h66 const RuntimeShape& input2_shape, in BinaryFunction()
Dstring_comparisons.h43 const RuntimeShape& input2_shape, in ComparisonStringImpl()
Dreference_ops.h224 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
243 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
272 const RuntimeShape& input2_shape, const int16_t* input2_data, in Sub16()
1168 const RuntimeShape& input2_shape, const T3* input2_data, in ArgMax()
1348 const RuntimeShape& input2_shape, const T* input2_data, in Pow()
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c773 const size_t* input2_shape, in setup_binary_elementwise_nd()
963 const size_t* input2_shape, in setup_binary_elementwise_nd_f16()
994 const size_t* input2_shape, in setup_binary_elementwise_nd_f32()
1019 const size_t* input2_shape, in xnn_setup_add_nd_qs8()
1043 const size_t* input2_shape, in xnn_setup_add_nd_qu8()
1067 const size_t* input2_shape, in xnn_setup_add_nd_f16()
1087 const size_t* input2_shape, in xnn_setup_add_nd_f32()
1107 const size_t* input2_shape, in xnn_setup_divide_nd_f32()
1127 const size_t* input2_shape, in xnn_setup_maximum_nd_f32()
1147 const size_t* input2_shape, in xnn_setup_minimum_nd_f32()
[all …]
/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/gradient/
Dbcast_grad_args.cc52 const RuntimeShape input2_shape = GetTensorShape(input2); in Prepare() local
75 const RuntimeShape input2_shape = GetTensorShape(input2); in Invoke() local
/external/tensorflow/tensorflow/lite/kernels/
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()
Dlogical_test.cc32 std::initializer_list<int> input2_shape, BuiltinOperator op) { in LogicalOpModel()
Dadd_test.cc44 const std::vector<int>& input2_shape, in BaseAddOpModel()
Dcomparisons_test.cc36 std::initializer_list<int> input2_shape, in ComparisonOpModel()
/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/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/
Doptimized_ops.h1848 const RuntimeShape& input2_shape, const float* input2_data, in Add()
2085 const RuntimeShape& input2_shape, const uint8* input2_data, in Add()
2102 const RuntimeShape& input2_shape, const int16* input2_data, in Add()
2140 const T* input1_data, const RuntimeShape& input2_shape, in Add()
2174 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastAddDispatch()
2273 const RuntimeShape& input2_shape, const float* input2_data, in Mul()
2284 const RuntimeShape& input2_shape, const int32* input2_data, in Mul()
2302 const RuntimeShape& input2_shape, in MulNoActivation()
2328 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
2349 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h899 const RuntimeShape& input2_shape, NdArrayDesc<N>* desc0_out, in NdArrayDescsForElementwiseBroadcast()