| /external/gemmlowp/meta/ |
| D | streams.h | 47 template <typename InType>
48 class StreamUtil<InType, RowMajor> {
50 static const InType* Offset(const RowMajor& params, const InType* source, in Offset()
52 return reinterpret_cast<const InType*>( in Offset()
54 offset_stride * params.stride + offset_advance * sizeof(InType)); in Offset()
57 static InType* Offset(const RowMajor& params, InType* source, in Offset()
59 return reinterpret_cast<InType*>(reinterpret_cast<std::uint8_t*>(source) + in Offset()
61 offset_advance * sizeof(InType)); in Offset()
69 template <typename InType>
70 class StreamUtil<InType, RowMajorWithSum> {
[all …]
|
| D | transform_kernels.h | 70 template <typename InType, typename OutType, int kernel_size, int leftovers>
71 class Transform1DKernel<InType, OutType, Quantize, kernel_size, leftovers> {
73 static void Transform(const InType* in, const Quantize& params, in Transform()
77 std::cout << "Quantize::Transform(" << std::string(typeid(InType).name()) in Transform()
88 template <typename InType, typename OutType, int kernel_size, int leftovers>
89 class Transform1DKernel<InType, OutType, Dequantize, kernel_size, leftovers> {
91 static void Transform(const InType* in, const Dequantize& params, in Transform()
95 std::cout << "Dequantize::Transform(" << std::string(typeid(InType).name()) in Transform()
106 template <typename InType, typename OutType, int kernel_size, int leftovers>
107 class Transform1DKernel<InType, OutType, Requantize, kernel_size, leftovers> {
[all …]
|
| D | base.h | 49 typedef InType_ InType; typedef
64 const InType* lhs;
65 const InType* rhs;
76 template <typename InType, int lanes_count, int pack_size, int leftovers,
80 static void Pack(const InType* in, const StreamParams& params, InType* out);
91 template <typename InType, typename StreamType>
94 static const InType* Offset(const StreamType& params, const InType* source,
100 template <typename InType, typename OutType, typename Kernel,
104 static void Multiply(const InType* lhs, const InType* rhs,
111 typedef InType_ InType; typedef
[all …]
|
| D | single_thread_gemm.h | 34 StreamUtil<typename P::InType, typename P::LeftStream>::Scratch( in EstimateScratchSize()
39 StreamUtil<typename P::InType, typename P::RightStream>::Scratch( in EstimateScratchSize()
48 typedef typename P::InType InType; in ExecuteDispatch3D() typedef
51 typedef Stream<typename P::InType, m, k, k_leftovers, in ExecuteDispatch3D()
54 typedef Stream<typename P::InType, m_leftovers, k, k_leftovers, in ExecuteDispatch3D()
58 typedef Stream<typename P::InType, n, k, k_leftovers, in ExecuteDispatch3D()
61 typedef Stream<typename P::InType, n_leftovers, k, k_leftovers, 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()
[all …]
|
| D | quantized_mul_kernels.h | 35 template <typename InType, typename OutType, int m, int n, int k>
36 class MulKernel<InType, OutType, QuantizedStaticPreprocessed, RowMajor, m, n,
41 static void Multiply(const InType* lhs, const InType*, in Multiply() argument
45 std::cout << "MulQSPR(" << typeid(InType).name() << ", " in Multiply()
61 std::cout << "MulQSPR(" << typeid(InType).name() << ", " in Debug()
83 template <typename InType, typename OutType, int m, int n, int k>
84 class MulKernel<InType, OutType, QuantizedStaticPreprocessedAsInt32, RowMajor,
90 static void Multiply(const InType* lhs, const InType*, in Multiply() argument
94 std::cout << "MulQSPI32R(" << typeid(InType).name() << ", " in Multiply()
110 std::cout << "MulQSPI32R(" << typeid(InType).name() << ", " in Debug()
[all …]
|
| D | multi_thread_gemm.h | 39 StreamUtil<typename Params::InType, typename Params::LeftStream>::Offset( in PrepareGemmTask()
44 StreamUtil<typename Params::InType, typename Params::RightStream>::Offset( in PrepareGemmTask()
|
| D | single_thread_transform.h | 33 Transform1DKernel<typename P::InType, typename P::OutType, in ExecuteDispatch1D()
|
| D | multi_thread_transform.h | 32 typedef Transform1DUtil<typename Params::InType, typename Params::OutType, in PrepareTransform1DTasks()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | bfloat16.cc | 492 template <typename InType, typename OutType, typename Functor>
499 InType x = *reinterpret_cast<const InType*>(i0); in BinaryUFunc()
500 InType y = *reinterpret_cast<const InType*>(i1); in BinaryUFunc()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 1871 template <typename InType>
1879 std::is_same<InType, Eigen::half>::value) { in TensorOpsAvailable()
|