Home
last modified time | relevance | path

Searched refs:ElementwiseBinaryFunction (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DElementwiseFunction.cpp23 ElementwiseBinaryFunction<Functor>::ElementwiseBinaryFunction(const TensorShape& inShape0, in ElementwiseBinaryFunction() function in armnn::ElementwiseBinaryFunction
64 template struct armnn::ElementwiseBinaryFunction<std::plus<float>>; member in armnn
65 template struct armnn::ElementwiseBinaryFunction<std::minus<float>>; member in armnn
66 template struct armnn::ElementwiseBinaryFunction<std::multiplies<float>>; member in armnn
67 template struct armnn::ElementwiseBinaryFunction<std::divides<float>>; member in armnn
68 template struct armnn::ElementwiseBinaryFunction<armnn::maximum<float>>; member in armnn
69 template struct armnn::ElementwiseBinaryFunction<armnn::minimum<float>>; member in armnn
71 template struct armnn::ElementwiseBinaryFunction<std::plus<int32_t>>; member in armnn
72 template struct armnn::ElementwiseBinaryFunction<std::minus<int32_t>>; member in armnn
73 template struct armnn::ElementwiseBinaryFunction<std::multiplies<int32_t>>; member in armnn
[all …]
H A DRefElementwiseBinaryWorkload.cpp41 using AddFunction = ElementwiseBinaryFunction<std::plus<DataType>>; in ExecuteFunction()
42 using DivFunction = ElementwiseBinaryFunction<std::divides<DataType>>; in ExecuteFunction()
43 using MaximumFunction = ElementwiseBinaryFunction<armnn::maximum<DataType>>; in ExecuteFunction()
44 using MinimumFunction = ElementwiseBinaryFunction<armnn::minimum<DataType>>; in ExecuteFunction()
45 using MulFunction = ElementwiseBinaryFunction<std::multiplies<DataType>>; in ExecuteFunction()
46 using SubFunction = ElementwiseBinaryFunction<std::minus<DataType>>; in ExecuteFunction()
H A DRefComparisonWorkload.cpp74 using EqualFunction = ElementwiseBinaryFunction<std::equal_to<InType>>; in Execute()
75 using GreaterFunction = ElementwiseBinaryFunction<std::greater<InType>>; in Execute()
76 using GreaterOrEqualFunction = ElementwiseBinaryFunction<std::greater_equal<InType>>; in Execute()
77 using LessFunction = ElementwiseBinaryFunction<std::less<InType>>; in Execute()
78 using LessOrEqualFunction = ElementwiseBinaryFunction<std::less_equal<InType>>; in Execute()
79 using NotEqualFunction = ElementwiseBinaryFunction<std::not_equal_to<InType>>; in Execute()
H A DElementwiseFunction.hpp15 struct ElementwiseBinaryFunction struct
20 ElementwiseBinaryFunction(const TensorShape& inShape0,
H A DRefElementwiseWorkload.hpp29 using InType = typename ElementwiseBinaryFunction<Functor>::InType;
30 using OutType = typename ElementwiseBinaryFunction<Functor>::OutType;
H A DRefElementwiseWorkload.cpp59 ElementwiseBinaryFunction<Functor>(inShape0, in Execute()