Home
last modified time | relevance | path

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

12

/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()
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 …]
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()
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()
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.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()
Dbroadcast_args.h26 const RuntimeShape& input2_shape, const T* input2_data, in BroadcastArgs()
Dstring_comparisons.h43 const RuntimeShape& input2_shape, in ComparisonStringImpl()
Dbinary_function.h77 const RuntimeShape& input2_shape, in BinaryFunction()
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()
/external/XNNPACK/src/operators/
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 …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
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()
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()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
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()
Dmul.h234 const RuntimeShape& input2_shape, const int8* input2_data, in Mul()
248 const RuntimeShape& input2_shape, in BroadcastMulDispatch()
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()
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args.cc53 const RuntimeShape input2_shape = GetTensorShape(input2); in Prepare() local
76 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.cc45 const std::vector<int>& input2_shape, in BaseAddOpModel()
Dcomparisons_test.cc36 std::initializer_list<int> input2_shape, in ComparisonOpModel()
/external/XNNPACK/test/
Dsubgraph-binary-tester.h44 std::vector<size_t> input2_shape; in SetUp() local
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
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
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 …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h996 const RuntimeShape& input2_shape, NdArrayDesc<N>* desc0_out, in NdArrayDescsForElementwiseBroadcast()
/external/pytorch/aten/src/ATen/test/
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