Home
last modified time | relevance | path

Searched refs:OutType (Results 1 – 25 of 41) sorted by relevance

12

/external/gemmlowp/meta/
Dtransform_kernels.h70 template <typename InType, typename OutType, int kernel_size, int leftovers>
71 class Transform1DKernel<InType, OutType, Quantize, kernel_size, leftovers> {
74 OutType* output) { in Transform()
78 << ", " << std::string(typeid(OutType).name()) << ") -- " in Transform()
88 template <typename InType, typename OutType, int kernel_size, int leftovers>
89 class Transform1DKernel<InType, OutType, Dequantize, kernel_size, leftovers> {
92 OutType* output) { in Transform()
96 << ", " << std::string(typeid(OutType).name()) << ") -- " in Transform()
106 template <typename InType, typename OutType, int kernel_size, int leftovers>
107 class Transform1DKernel<InType, OutType, Requantize, kernel_size, leftovers> {
[all …]
Dquantized_mul_kernels.h35 template <typename InType, typename OutType, int m, int n, int k>
36 class MulKernel<InType, OutType, QuantizedStaticPreprocessed, RowMajor, m, n,
42 const FusedKernel& params, OutType* result) { in Multiply()
46 << typeid(OutType).name() << ")::Multiply() -- " << m << "x" << n in Multiply()
62 << typeid(OutType).name() << ") -- " << m << "x" << n << "x" << k in Debug()
83 template <typename InType, typename OutType, int m, int n, int k>
84 class MulKernel<InType, OutType, QuantizedStaticPreprocessedAsInt32, RowMajor,
91 const FusedKernel& params, OutType* result) { in Multiply()
95 << typeid(OutType).name() << ")::Multiply() -- " << m << "x" << n in Multiply()
111 << typeid(OutType).name() << ") -- " << m << "x" << n << "x" << k in Debug()
[all …]
Dbase.h50 typedef OutType_ OutType; typedef
66 OutType* result;
100 template <typename InType, typename OutType, typename Kernel,
106 OutType* result);
112 typedef OutType_ OutType; typedef
116 OutType* output;
122 template <typename InType, typename OutType, typename Kernel, int kernel_size,
127 OutType* output);
130 template <typename InType, typename OutType, typename Transform>
138 static OutType* OffsetOutput(const Transform& params, OutType* output,
Dsingle_thread_gemm.h49 typedef typename P::OutType OutType; in ExecuteDispatch3D() typedef
65 typedef Stream<typename P::OutType, m, n, 0, typename P::OutputStream> in ExecuteDispatch3D()
67 typedef Stream<typename P::OutType, m_leftovers, n, 0, in ExecuteDispatch3D()
71 typedef MulKernel<typename P::InType, typename P::OutType, in ExecuteDispatch3D()
74 typedef MulKernel<typename P::InType, typename P::OutType, in ExecuteDispatch3D()
78 typedef MulKernel<typename P::InType, typename P::OutType, in ExecuteDispatch3D()
82 typedef MulKernel<typename P::InType, typename P::OutType, in ExecuteDispatch3D()
171 reinterpret_cast<OutType*>(result_chunk)); in ExecuteDispatch3D()
180 reinterpret_cast<OutType*>(result_chunk)); in ExecuteDispatch3D()
203 reinterpret_cast<OutType*>(result_chunk)); in ExecuteDispatch3D()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dsearchsorted_op_gpu.cu.cc33 template <typename T, typename OutType>
37 OutType* __restrict__ outputs) { in UpperBoundKernel()
41 outputs[work_unit_id] = gpu_helper::upper_bound<T, OutType>( in UpperBoundKernel()
46 template <typename T, typename OutType>
50 OutType* __restrict__ outputs) { in LowerBoundKernel()
54 outputs[work_unit_id] = gpu_helper::lower_bound<T, OutType>( in LowerBoundKernel()
61 template <typename T, typename OutType>
62 struct UpperBoundFunctor<GPUDevice, T, OutType> {
67 typename TTypes<OutType, 1>::Tensor* output) { in Compute()
76 UpperBoundKernel<T, OutType>, config.block_count, in Compute()
[all …]
Dsearchsorted_op.cc36 template <typename T, typename OutType>
37 struct UpperBoundFunctor<CPUDevice, T, OutType> {
42 typename TTypes<OutType, 1>::Tensor* output) { in Compute()
46 OutType* output_ptr = output->data() + b * num_values; in Compute()
65 template <typename T, typename OutType>
66 struct LowerBoundFunctor<CPUDevice, T, OutType> {
71 typename TTypes<OutType, 1>::Tensor* output) { in Compute()
75 OutType* output_ptr = output->data() + b * num_values; in Compute()
95 template <typename Device, typename T, typename OutType>
130 auto output = output_t->template flat<OutType>(); in Compute()
[all …]
Dsearchsorted_op.h28 template <typename Device, typename T, typename OutType>
36 typename TTypes<OutType, 1>::Tensor* output);
39 template <typename Device, typename T, typename OutType>
47 typename TTypes<OutType, 1>::Tensor* output);
Dshape_ops.h41 template <typename OutType>
52 auto vec = out->vec<OutType>(); in Compute()
61 vec(i) = static_cast<OutType>(dim_size); in Compute()
68 template <typename OutType>
80 auto vec = out->vec<OutType>(); in Compute()
90 vec(j) = static_cast<OutType>(dim_size); in Compute()
114 template <typename OutType>
131 out->scalar<OutType>()() = static_cast<OutType>(size); in Compute()
/external/armnn/src/backends/reference/workloads/
DElementwiseFunction.hpp17 using OutType = typename Functor::result_type; typedef
25 Encoder<OutType>& outData);
31 using OutType = typename Functor::result_type; typedef
37 Encoder<OutType>& outData);
43 using OutType = bool; typedef
51 Encoder<OutType>& outData);
57 using OutType = bool; typedef
63 Encoder<OutType>& outData);
DElementwiseFunction.cpp28 Encoder<OutType>& outData) in ElementwiseBinaryFunction()
37 Encoder<OutType>& outData) in ElementwiseUnaryFunction()
48 Encoder<OutType>& outData) in LogicalBinaryFunction()
57 Encoder<OutType>& outData) in LogicalUnaryFunction()
DRefComparisonWorkload.hpp30 using OutType = bool; typedef in armnn::RefComparisonWorkload
34 std::unique_ptr<Encoder<OutType>> m_Output;
DRefLogicalUnaryWorkload.cpp47 std::unique_ptr<Encoder<OutType>> output = MakeEncoder<OutType>(outputInfo, outputs[0]->Map()); in Execute()
DRefElementwiseWorkload.hpp30 using OutType = typename ElementwiseBinaryFunction<Functor>::OutType; typedef in armnn::RefElementwiseWorkload
DRefLogicalBinaryWorkload.cpp50 std::unique_ptr<Encoder<OutType>> output = MakeEncoder<OutType>(outputInfo, outputs[0]->Map()); in Execute()
DRefElementwiseUnaryWorkload.cpp57 std::unique_ptr<Encoder<OutType>> output= MakeEncoder<OutType>(outputInfo, outputs[0]->Map()); in Execute()
DRefElementwiseWorkload.cpp57 std::unique_ptr<Encoder<OutType>> output= MakeEncoder<OutType>(outputInfo, outputs[0]->Map()); in Execute()
DRefLogicalBinaryWorkload.hpp28 using OutType = bool; typedef in armnn::RefLogicalBinaryWorkload
DRefLogicalUnaryWorkload.hpp28 using OutType = bool; typedef in armnn::RefLogicalUnaryWorkload
DRefElementwiseUnaryWorkload.hpp28 using OutType = float; typedef in armnn::RefElementwiseUnaryWorkload
/external/angle/src/libANGLE/
Dqueryconversions.h84 template <typename OutType>
85 OutType ConvertGLenum(GLenum param)
87 return static_cast<OutType>(param);
90 template <typename InType, typename OutType>
91 void ConvertGLenumValue(InType param, OutType *out)
93 *out = ConvertGLenum<OutType>(static_cast<GLenum>(param));
96 template <typename PackedEnumType, typename OutType>
97 void ConvertPackedEnum(PackedEnumType param, OutType *out)
99 *out = static_cast<OutType>(ToGLenum(param));
/external/libaom/test/
Dtransform_test_base.h40 template <typename OutType>
46 virtual void RunFwdTxfm(const int16_t *in, OutType *out, int stride) = 0;
48 virtual void RunInvTxfm(const OutType *out, uint8_t *dst, int stride) = 0;
59 OutType *test_temp_block = reinterpret_cast<OutType *>( in RunAccuracyCheck()
136 OutType *output_ref_block = reinterpret_cast<OutType *>( in RunCoeffCheck()
139 OutType *output_block = reinterpret_cast<OutType *>( in RunCoeffCheck()
188 OutType *trans_block = reinterpret_cast<OutType *>( in RunInvCoeffCheck()
238 OutType *output_ref_block = reinterpret_cast<OutType *>( in RunMemCheck()
241 OutType *output_block = reinterpret_cast<OutType *>( in RunMemCheck()
283 OutType *coeff = reinterpret_cast<OutType *>( in RunInvAccuracyCheck()
[all …]
/external/armnn/src/backends/backendsCommon/test/layerTests/
DDequantizeTestImpl.cpp63 typename OutType=armnn::ResolveType<ArmnnOutputType>>
64 LayerTestResult<OutType, 4> DequantizeSimpleTest( in DequantizeSimpleTest()
83 std::vector<OutType> expectedOutputData; in DequantizeSimpleTest()
84 for (OutType i = OutType(1); i <= OutType(12); ++i) in DequantizeSimpleTest()
89 return DequantizeTestImpl<T, 4, OutType>(workloadFactory, in DequantizeSimpleTest()
/external/XNNPACK/bench/
Dutils.h37 template<class InType, class OutType>
48 const size_t elementwise_size = sizeof(InType) + sizeof(OutType); in UnaryElementwiseParameters()
58 template<class InType, class OutType>
69 const size_t elementwise_size = 2 * sizeof(InType) + sizeof(OutType); in BinaryElementwiseParameters()
/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper.h370 template <typename T, typename OutType = int32, typename Iterator = const T*>
371 __device__ OutType upper_bound(Iterator first, OutType count, T val) {
373 OutType step = 0;
392 template <typename T, typename OutType = int32, typename Iterator = const T*>
393 __device__ OutType lower_bound(Iterator first, OutType count, T val) {
395 OutType step = 0;
/external/tensorflow/tensorflow/lite/kernels/
Dshape.cc32 template <typename OutType>
33 void ExtractShape(const TfLiteTensor* input, OutType* output_data) { in ExtractShape()

12